Mungert commited on
Commit
30afd38
·
verified ·
1 Parent(s): 20c71bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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)