Sangjun2 commited on
Commit
e91c036
·
verified ·
1 Parent(s): 2aff141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def predict_model2(image):
126
  attention_mask = inputs.attention_mask.to(device)
127
 
128
  model2.eval()
129
- t5_model.eval()
130
  with torch.no_grad():
131
  deplot_generated_ids = model2.generate(flattened_patches=flattened_patches, attention_mask=attention_mask, max_length=1000)
132
  generated_datatable = processor2.batch_decode(deplot_generated_ids, skip_special_tokens=False)[0]
 
126
  attention_mask = inputs.attention_mask.to(device)
127
 
128
  model2.eval()
129
+
130
  with torch.no_grad():
131
  deplot_generated_ids = model2.generate(flattened_patches=flattened_patches, attention_mask=attention_mask, max_length=1000)
132
  generated_datatable = processor2.batch_decode(deplot_generated_ids, skip_special_tokens=False)[0]