Update app.py

app.py

@@ -56,9 +56,13 @@ def predict_disease(image):
     predicted_label = class_names[predicted_class.item()]
 
     # Generate AI response based on the detected disease
-    prompt = f"You are an expert in sugarcane farming. The detected disease is '{predicted_label}'. Provide advice for the farmer."
+    prompt = f"The detected sugarcane disease is '{predicted_label}'. Provide detailed advice for managing this condition."
     ai_response = ai_pipeline(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
-
+    
+    # Remove the prompt text if it appears in the AI response
+    if ai_response.startswith(prompt):
+        ai_response = ai_response[len(prompt):].strip()
+    
     # Create a styled HTML output
     output_message = f"""
     <div style='font-size: 18px; color: #4CAF50; font-weight: bold;'>