Update app.py

app.py

@@ -34,7 +34,7 @@ def extract_assistant_reply(input_string):
 
 
 @spaces.GPU
-def diagnose_router(image):
+def med_interpreter(image):
     messages = [
         {"role": "user", "content": [
             {"type": "image"},
@@ -45,7 +45,7 @@ def diagnose_router(image):
     inputs = processor(image, input_text, return_tensors="pt").to(model.device)
 
     # Generate the output from the model
-    output = model.generate(**inputs, max_new_tokens=300)
+    output = model.generate(**inputs, max_new_tokens=4000)
     print(output)
     markdown_text = processor.decode(output[0])
     print(markdown_text)
@@ -56,11 +56,11 @@ def diagnose_router(image):
 
 # Gradio UI
 interface = gr.Interface(
-    fn=diagnose_router,
-    inputs=gr.Image(type="pil", label="Upload an image of the faulty router"),
+    fn=med_interpreter,
+    inputs=gr.Image(type="pil", label="Upload an image of the medical report"),
     outputs=gr.HTML(),
-    title="Router Diagnosis",
-    description="Upload a photo of your router to receive a professional diagnosis and troubleshooting steps displayed in a structured, easy-to-read format."
+    title="Medical Report Insights",
+    description="Upload an image of your medical report to get the interperation of it"
 )
 
 # Launch the UI