Spaces:
Sleeping
Sleeping
Commit
·
2caa9f6
1
Parent(s):
e6e5fc7
rm: local path
Browse files- .gitignore +1 -0
- crop_utils.py +3 -6
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
__pycache__
|
crop_utils.py
CHANGED
@@ -9,9 +9,10 @@ import pyrebase
|
|
9 |
import requests
|
10 |
from openai import OpenAI
|
11 |
from PIL import Image, ImageDraw, ImageFont
|
12 |
-
from prompts import remove_unwanted_prompt
|
13 |
from ultralytics import YOLO
|
14 |
|
|
|
|
|
15 |
model = YOLO("yolo11n.pt")
|
16 |
|
17 |
|
@@ -851,11 +852,7 @@ def get_image_crop(cid=None, rsid=None, uid=None):
|
|
851 |
Returns:
|
852 |
gr.Gallery: Gallery of all generated images
|
853 |
"""
|
854 |
-
|
855 |
-
# image_paths = get_sprite_firebase(cid, rsid, uid)
|
856 |
-
|
857 |
-
# For testing, use a local image path
|
858 |
-
image_paths = ["sprite1.jpg", "sprite2.jpg"]
|
859 |
|
860 |
# Lists to store all images
|
861 |
all_images = []
|
|
|
9 |
import requests
|
10 |
from openai import OpenAI
|
11 |
from PIL import Image, ImageDraw, ImageFont
|
|
|
12 |
from ultralytics import YOLO
|
13 |
|
14 |
+
from prompts import remove_unwanted_prompt
|
15 |
+
|
16 |
model = YOLO("yolo11n.pt")
|
17 |
|
18 |
|
|
|
852 |
Returns:
|
853 |
gr.Gallery: Gallery of all generated images
|
854 |
"""
|
855 |
+
image_paths = get_sprite_firebase(cid, rsid, uid)
|
|
|
|
|
|
|
|
|
856 |
|
857 |
# Lists to store all images
|
858 |
all_images = []
|