appilcation should be finished

app.py

@@ -1,12 +1,22 @@
 import gradio as gr
-from transformers import AutoModel
+from ultralytics import YOLO
 
 
-def greet(name):
-    return "Hello " + name + "!!"
+def yoloV8_func(image: gr.inputs.Image = None):
+    model_path = "models/last.pt"
+    model = YOLO(model_path)
+    results = model.predict(image)
+    mask_json = results[0].masks.data.tojson()
 
-model = AutoModel.from_pretrained("models/last.pt")
-pipe = pipeline("image-segmentation", model = model)
-iface = gr.Interface.from_pipeline(pipe)
+    return str(mask_json)
+
+
+yolo_app = gr.Interface(
+    fn=yoloV8_func,
+    inputs=gr.inputs.Image(type="filepath", label="Input Image"),
+    outputs=gr.Textbox(),
+    title="YOLOv8: Custom Instance segmentation on lettuce leafs",
+    cache_examples=True,
+)
 # iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+yolo_app.launch()

requirements.txt

@@ -2,4 +2,4 @@ opencv-python-headless
 pandas
 ultralytics
 Pillow
-transformers
+gradio