Update app.py

app.py

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