Spaces:

nttdataspain
/

Image-To-Text-Lora-ViT

Runtime error

D0k-tor commited on Jun 19, 2023

Commit

751acdb

1 Parent(s): 2baef79

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)
 # Predict function
 def predict(image):
-    img = Image.open(image).convert('RGB')
     model.eval()
     pixel_values = feature_extractor(images=[img], return_tensors="pt").pixel_values
     with torch.no_grad():

 # Predict function
 def predict(image):
+    img = image.convert('RGB')
     model.eval()
     pixel_values = feature_extractor(images=[img], return_tensors="pt").pixel_values
     with torch.no_grad():