gokaygokay
commited on
Commit
•
9af016f
1
Parent(s):
25d724e
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def run_example(image):
|
|
48 |
if image.mode != "RGB":
|
49 |
image = image.convert("RGB")
|
50 |
|
51 |
-
inputs = processor(text=prompt, images=image, return_tensors="pt").to(
|
52 |
generated_ids = model.generate(
|
53 |
input_ids=inputs["input_ids"],
|
54 |
pixel_values=inputs["pixel_values"],
|
|
|
48 |
if image.mode != "RGB":
|
49 |
image = image.convert("RGB")
|
50 |
|
51 |
+
inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
|
52 |
generated_ids = model.generate(
|
53 |
input_ids=inputs["input_ids"],
|
54 |
pixel_values=inputs["pixel_values"],
|