Update app.py

app.py

@@ -9,9 +9,10 @@ client = InferenceClient()
 # Function to analyze plant images
 def analyze_plant_image(image):
     # Save image to a temporary location with appropriate extension
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_image:
-        image.save(temp_image.name, format="JPEG")
-        image_url = f"file://{temp_image.name}"
+    temp_dir = tempfile.mkdtemp()
+    temp_image_path = os.path.join(temp_dir, "uploaded_image.jpg")
+    image.save(temp_image_path, format="JPEG")
+    image_url = f"file://{temp_image_path}"
 
     # Create the message structure
     messages = [
@@ -36,7 +37,7 @@ def analyze_plant_image(image):
     stream = client.chat.completions.create(
         model="Qwen/Qwen2-VL-7B-Instruct",
         messages=messages,
-        max_tokens=500,
+        max_tokens=5000,
         stream=True
     )
 
@@ -54,7 +55,7 @@ with gr.Blocks() as app:
     with gr.Row():
         # First column for input components
         with gr.Column():
-            image_input = gr.Image(type="pil", label="Upload Plant Image", image_mode="RGB")
+            image_input = gr.Image(type="pil", label="Upload Plant Image", image_mode="RGB", tool="editor")
             analyze_button = gr.Button("Analyze Plant Image")
 
         # Second column for output
@@ -65,4 +66,4 @@ with gr.Blocks() as app:
     analyze_button.click(fn=analyze_plant_image, inputs=image_input, outputs=output_markdown)
 
 # Run the Gradio app
-app.launch()
+app.launch()