print

app.py

@@ -130,8 +130,10 @@ def diagnose_plant_from_image(uploaded_image: Image.Image) -> str:
         outputs = VISION_MODEL.generate(**inputs, max_new_tokens=48, use_cache=True)
 
     response = PROCESSOR.batch_decode(outputs, skip_special_tokens=True)[0]
+    print(response)
     # Clean up the model's output to get only the diagnosis
     answer_start_index = response.rfind("model\n")
+    print(answer_start_index)
     if answer_start_index != -1:
         return response[answer_start_index + len("model\n"):].strip()
     return "Could not parse diagnosis from model output."