Update app.py

app.py

@@ -46,9 +46,9 @@ 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
+    index = decoded_output.lower().find("model")
+    if index != -1:
+        return decoded_output[index + len("model"):].strip()
 
     # Fallback: return full decoded output if structure is unexpected
     return decoded_output