Alex Lopez
commited on
Commit
·
0c30ec5
1
Parent(s):
153eee4
modify requirements.txt
Browse files- README.md +1 -1
- app.py +4 -3
- requirements.txt +2 -2
README.md
CHANGED
|
@@ -4,7 +4,7 @@ emoji: 🚀
|
|
| 4 |
colorFrom: gray
|
| 5 |
colorTo: green
|
| 6 |
sdk: gradio
|
| 7 |
-
sdk_version: 5.
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
|
|
|
| 4 |
colorFrom: gray
|
| 5 |
colorTo: green
|
| 6 |
sdk: gradio
|
| 7 |
+
sdk_version: 5.26.0
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
app.py
CHANGED
|
@@ -2,10 +2,7 @@ import gradio as gr
|
|
| 2 |
from ultralytics import YOLO
|
| 3 |
import cv2
|
| 4 |
import numpy as np
|
| 5 |
-
from PIL import Image
|
| 6 |
|
| 7 |
-
# Load the YOLO model with the provided weights
|
| 8 |
-
model = YOLO("yolo_weigths.pt")
|
| 9 |
|
| 10 |
def segment_image(input_image):
|
| 11 |
"""
|
|
@@ -32,6 +29,10 @@ def segment_image(input_image):
|
|
| 32 |
|
| 33 |
return annotated_img
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
# Create Gradio interface
|
| 36 |
demo = gr.Interface(
|
| 37 |
fn=segment_image,
|
|
|
|
| 2 |
from ultralytics import YOLO
|
| 3 |
import cv2
|
| 4 |
import numpy as np
|
|
|
|
| 5 |
|
|
|
|
|
|
|
| 6 |
|
| 7 |
def segment_image(input_image):
|
| 8 |
"""
|
|
|
|
| 29 |
|
| 30 |
return annotated_img
|
| 31 |
|
| 32 |
+
|
| 33 |
+
# Load the YOLO model with the provided weights
|
| 34 |
+
model = YOLO("yolo_weigths.pt")
|
| 35 |
+
|
| 36 |
# Create Gradio interface
|
| 37 |
demo = gr.Interface(
|
| 38 |
fn=segment_image,
|
requirements.txt
CHANGED
|
@@ -3,8 +3,8 @@ opencv-python
|
|
| 3 |
matplotlib
|
| 4 |
loguru
|
| 5 |
ultralytics
|
| 6 |
-
gradio[mcp]
|
| 7 |
mcp
|
| 8 |
-
|
| 9 |
|
| 10 |
|
|
|
|
| 3 |
matplotlib
|
| 4 |
loguru
|
| 5 |
ultralytics
|
| 6 |
+
# gradio[mcp]
|
| 7 |
mcp
|
| 8 |
+
https://gradio-pypi-previews.s3.amazonaws.com/3b5cace94781b90993b596a83fb39fd1584d68ee/gradio-5.26.0-py3-none-any.whl
|
| 9 |
|
| 10 |
|