Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def predict_model2(image):
|
|
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]
|
|
|
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]
|