Spaces:

ydshieh
/

Kosmos-2

Running on T4

ydshieh HF staff commited on Oct 31, 2023

Commit

02572c5

•

1 Parent(s): 185d9a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -186,7 +186,7 @@ def main():
         else:
             text_input = f"<grounding>{text_input}"
-        inputs = processor(text=text_input, images=image_input, return_tensors="pt")
         generated_ids = model.generate(
             pixel_values=inputs["pixel_values"],

         else:
             text_input = f"<grounding>{text_input}"
+        inputs = processor(text=text_input, images=image_input, return_tensors="pt").to("cuda")
         generated_ids = model.generate(
             pixel_values=inputs["pixel_values"],