Florence-2-SD3-Captioner

Running on Zero

gokaygokay commited on Jun 24

Commit

9af016f

•

1 Parent(s): 25d724e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def run_example(image):
     if image.mode != "RGB":
         image = image.convert("RGB")
-    inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],

     if image.mode != "RGB":
         image = image.convert("RGB")
+    inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],