Spaces:

microsoft
/

document-image-transformer

Running

nielsr HF Staff commited on Mar 8, 2022

Commit

dbd7df0

1 Parent(s): 9c08f83

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 import gradio as gr
 import torch
-feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/beit-base-finetuned-rvlcdip")
-model = AutoModelForImageClassification.from_pretrained("microsoft/beit-base-finetuned-rvlcdip")
 def classify_image(image):
    pixel_values = feature_extractor(image, return_tensors="pt")

 import gradio as gr
 import torch
+feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/dit-base-finetuned-rvlcdip")
+model = AutoModelForImageClassification.from_pretrained("microsoft/dit-base-finetuned-rvlcdip")
 def classify_image(image):
    pixel_values = feature_extractor(image, return_tensors="pt")