Update app.py

app.py

@@ -32,7 +32,13 @@ def generate_code(prompt):
             temperature=0.7
         )
 
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    # Strip the prompt if it appears at the start
+    if output_text.startswith(prompt):
+        output_text = output_text[len(prompt):].lstrip()
+
+    return output_text
 
 demo = gr.Interface(
     fn=generate_code,