try again

app.py

@@ -49,15 +49,11 @@ def predict(code):
         top_p=0.09,
         temperature=1.2,
         repetition_penalty=1.1,
+        return_full_text=False
     )
     print(f"Pipe out: {repr(pipe_out)}")
 
-    return tokenizer.decode(
-        model.generate(
-            tokenizer.encode(prompt, return_tensors="pt"), max_length=10**6
-        )[0]
-    )
-
+    return pipe_out
 
 demo = gr.Interface(
     fn=predict,