musadac commited on
Commit
be1456b
1 Parent(s): 4f2b696

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ if uploaded_file is not None:
85
  pixel_values = processortext2(img.convert("RGB"), return_tensors="pt").pixel_values
86
 
87
  with torch.no_grad():
88
- generated_ids = model2.generate(img_tensor)
89
 
90
  result = processortext2.batch_decode(generated_ids, skip_special_tokens=True)[0]
91
  st.write("OCR Result:")
 
85
  pixel_values = processortext2(img.convert("RGB"), return_tensors="pt").pixel_values
86
 
87
  with torch.no_grad():
88
+ generated_ids = model2.generate(pixel_values)
89
 
90
  result = processortext2.batch_decode(generated_ids, skip_special_tokens=True)[0]
91
  st.write("OCR Result:")