Florence-2-SD3-Captioner

Running on Zero

gokaygokay commited on Jun 24

Commit

c569a26

•

1 Parent(s): 9af016f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,8 @@ def modify_caption(caption: str) -> str:
 @spaces.GPU
 def run_example(image):
     image = Image.fromarray(image)
-    prompt = "<DESCRIPTION>" + "Describe this image in great detail."
     # Ensure the image is in RGB mode
     if image.mode != "RGB":

 @spaces.GPU
 def run_example(image):
     image = Image.fromarray(image)
+    task_prompt = "<DESCRIPTION>"
+    prompt = task_prompt + "Describe this image in great detail."
     # Ensure the image is in RGB mode
     if image.mode != "RGB":