Spaces:
Runtime error
Runtime error
humblemikey
commited on
Commit
•
b129969
1
Parent(s):
5763a51
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def process_image(image):
|
|
44 |
|
45 |
prompt = "<MORE_DETAILED_CAPTION>"
|
46 |
|
47 |
-
inputs = processor(text=prompt, images=image, return_tensors="pt")
|
48 |
generated_ids = model.generate(
|
49 |
input_ids=inputs["input_ids"],
|
50 |
pixel_values=inputs["pixel_values"],
|
|
|
44 |
|
45 |
prompt = "<MORE_DETAILED_CAPTION>"
|
46 |
|
47 |
+
inputs = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
|
48 |
generated_ids = model.generate(
|
49 |
input_ids=inputs["input_ids"],
|
50 |
pixel_values=inputs["pixel_values"],
|