Spaces:

humblemikey
/

thwri-CogFlorence-2

Runtime error

humblemikey commited on Jul 30, 2024

Commit

b129969

verified ·

1 Parent(s): 5763a51

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def process_image(image):
     prompt = "<MORE_DETAILED_CAPTION>"
-    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"],

     prompt = "<MORE_DETAILED_CAPTION>"
+    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"],