Update app.py

app.py

@@ -12,7 +12,7 @@ def predict(img):
   #   return generated_text
   # else:
 
-    images = img.convert("RGB")
+    images = Image.fromarray(img).convert("RGB")
     pixel_values = processor(images, return_tensors="pt").pixel_values
     # PIL_image = Image.fromarray(np.uint8(pixel_values.squeeze()[0,:,:])).convert('RGB')
     generated_ids = model.generate(pixel_values,max_length=512)