Update app.py

app.py

@@ -79,11 +79,12 @@ from fastsam import FastSAM, FastSAMPrompt
 
 model = FastSAM('./FastSAM.pt')
 DEVICE = 'cpu'
-IMAGE_PATH = 'sam.jpg'
 def segment_solar_panel(img):
-    img.Save(IMAGE_PATH)
-    everything_results = model(IMAGE_PATH, device=DEVICE, retina_masks=True, imgsz=1024, conf=0.4, iou=0.9,)
-    prompt_process = FastSAMPrompt(IMAGE_PATH, everything_results, device=DEVICE)
+    # os.system('python Inference.py --model_path FastSAM.pt --img_path bus.jpg  --text_prompt "solar panel grids"')
+    img = img.convert("RGB")
+
+    everything_results = model(img, device=DEVICE, retina_masks=True, imgsz=1024, conf=0.4, iou=0.9,)
+    prompt_process = FastSAMPrompt(img, everything_results, device=DEVICE)
     
     # everything prompt
     ann = prompt_process.everything_prompt()
@@ -99,8 +100,8 @@ def segment_solar_panel(img):
     # point_label default [0] [1,0] 0:background, 1:foreground
     ann = prompt_process.point_prompt(points=[[620, 360]], pointlabel=[1])
     
-    prompt_process.plot(annotations=ann,output_path='./dog.jpg',)
-    return Image.Open('./dog.jpg')
+    prompt_process.plot(annotations=ann,output_path='./bus.jpg',)
+    return Image.Open('./bus.jpg')
 
 
 import gradio as gr