Update app.py
Browse files
app.py
CHANGED
|
@@ -158,8 +158,7 @@ def process(image, instruction):
|
|
| 158 |
tokenizer,
|
| 159 |
data_processor,
|
| 160 |
use_placeholder=True,
|
| 161 |
-
topk=3
|
| 162 |
-
device=str(device),
|
| 163 |
)
|
| 164 |
except Exception as e:
|
| 165 |
print("inference error:", e)
|
|
|
|
| 158 |
tokenizer,
|
| 159 |
data_processor,
|
| 160 |
use_placeholder=True,
|
| 161 |
+
topk=3
|
|
|
|
| 162 |
)
|
| 163 |
except Exception as e:
|
| 164 |
print("inference error:", e)
|