Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 from utils.tools_gradio import fast_process
 from utils.tools import format_results, box_prompt, point_prompt, text_prompt
-from PIL import ImageDraw
+from PIL import ImageDraw,Image
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
@@ -61,7 +61,7 @@ def segment_everything(
         print(f"Largest Box {i+1}: {box.tolist()}")
     
     fig, ax = plt.subplots(1)
-    ax.imshow(image)
+    ax.imshow(input)
 
     for box in largest_boxes:
         x1, y1, x2, y2 = box[:4]