AhmadMustafa commited on
Commit
2caa9f6
·
1 Parent(s): e6e5fc7

rm: local path

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. 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
- # Uncomment this line when using Firebase
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 = []