Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,8 +35,8 @@ def preprocess_image(image):
|
|
| 35 |
# Load model directly
|
| 36 |
from transformers import AutoTokenizer, AutoModel,ViTFeatureExtractor,TrOCRProcessor,VisionEncoderDecoderModel
|
| 37 |
|
| 38 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 39 |
-
model1 = VisionEncoderDecoderModel.from_pretrained("
|
| 40 |
feature_extractor1 = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
|
| 41 |
|
| 42 |
processor1 = TrOCRProcessor(feature_extractor=feature_extractor1, tokenizer=tokenizer)
|
|
|
|
| 35 |
# Load model directly
|
| 36 |
from transformers import AutoTokenizer, AutoModel,ViTFeatureExtractor,TrOCRProcessor,VisionEncoderDecoderModel
|
| 37 |
|
| 38 |
+
tokenizer = AutoTokenizer.from_pretrained("aayushpuri01/TrOCR-Devanagari")
|
| 39 |
+
model1 = VisionEncoderDecoderModel.from_pretrained("aayushpuri01/TrOCR-Devanagari")
|
| 40 |
feature_extractor1 = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
|
| 41 |
|
| 42 |
processor1 = TrOCRProcessor(feature_extractor=feature_extractor1, tokenizer=tokenizer)
|