baohuynhbk14 commited on
Commit
2285967
·
1 Parent(s): 15d7f87

Reorder parameters in predict function to place state before image_path

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -185,8 +185,8 @@ model = AutoModel.from_pretrained(
185
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, use_fast=False)
186
 
187
  @spaces.GPU
188
- def predict(image_path,
189
- state,
190
  max_input_tiles=6,
191
  temperature=1.0,
192
  max_output_tokens=700,
 
185
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, use_fast=False)
186
 
187
  @spaces.GPU
188
+ def predict(state,
189
+ image_path,
190
  max_input_tiles=6,
191
  temperature=1.0,
192
  max_output_tokens=700,