abdulsamad commited on
Commit
a4d1c46
·
1 Parent(s): 48e5761

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -8,23 +8,23 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
8
  file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
9
 
10
  # Display the uploaded image
11
- if file is not None:
12
 
13
  # Read the image data
14
- image = Image.open(file)
15
  #image=image.convert("RGB")
16
 
17
  #resized_image = image.resize((384, 384))
18
 
19
- processor = TrOCRProcessor.from_pretrained('microsoft/trocr-small-handwritten')
20
- model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-small-handwritten')
21
- pixel_values = processor(images=resized_image, return_tensors="pt").pixel_values
22
- generated_ids = model.generate(pixel_values)
23
- generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
24
 
25
  # Display the image
26
- st.image(image, caption='Uploaded Image', use_column_width=True)
27
 
28
  # Print the shape of the image
29
- st.write(f"Text: {generated_text}")
30
 
 
8
  file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
9
 
10
  # Display the uploaded image
11
+ #if file is not None:
12
 
13
  # Read the image data
14
+ #image = Image.open(file)
15
  #image=image.convert("RGB")
16
 
17
  #resized_image = image.resize((384, 384))
18
 
19
+ #processor = TrOCRProcessor.from_pretrained('microsoft/trocr-small-handwritten')
20
+ #model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-small-handwritten')
21
+ #pixel_values = processor(images=resized_image, return_tensors="pt").pixel_values
22
+ #generated_ids = model.generate(pixel_values)
23
+ #generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
24
 
25
  # Display the image
26
+ #st.image(image, caption='Uploaded Image', use_column_width=True)
27
 
28
  # Print the shape of the image
29
+ #st.write(f"Text: {generated_text}")
30