Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
dec222b
1
Parent(s):
9d4c8e1
upg
Browse files
app.py
CHANGED
|
@@ -103,6 +103,9 @@ def process_image(image):
|
|
| 103 |
return_tensors="pt",
|
| 104 |
).to(device)
|
| 105 |
|
|
|
|
|
|
|
|
|
|
| 106 |
generation_config = {
|
| 107 |
"do_sample": False,
|
| 108 |
"num_beams": 1,
|
|
|
|
| 103 |
return_tensors="pt",
|
| 104 |
).to(device)
|
| 105 |
|
| 106 |
+
# Move model to GPU
|
| 107 |
+
model.to(device)
|
| 108 |
+
|
| 109 |
generation_config = {
|
| 110 |
"do_sample": False,
|
| 111 |
"num_beams": 1,
|