Update app.py

app.py

@@ -45,6 +45,12 @@ def generate_text(user_input):
     
     # Decode the model output and return the result
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
+
+    if "model:" in decoded_output.lower():
+        parts = decoded_output.split("model:", 1)
+        return parts[1].strip()  # Only return the model's reply
+
+    # Fallback: return full decoded output if structure is unexpected
     return decoded_output
 
 # Build the Gradio interface