mkoot007 commited on
Commit
63d09a4
1 Parent(s): 821d791

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import easyocr
5
  from transformers import pipeline
6
 
7
  # Create a text2text-generation pipeline with the "t5-small" model
8
- pipe = pipeline("text2text-generation", model="t5-small")
9
 
10
  # Initialize the EasyOCR reader for text extraction from images
11
  ocr_reader = easyocr.Reader(['en'])
 
5
  from transformers import pipeline
6
 
7
  # Create a text2text-generation pipeline with the "t5-small" model
8
+ pipe = pipeline("text2text-generation", model="t5-base")
9
 
10
  # Initialize the EasyOCR reader for text extraction from images
11
  ocr_reader = easyocr.Reader(['en'])