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)
|