Update app.py

app.py

@@ -27,19 +27,20 @@ feature_extractor = ViTFeatureExtractor.from_pretrained(repo_name)
 tokenizer = AutoTokenizer.from_pretrained(repo_name)
 model = VisionEncoderDecoderModel.from_pretrained(repo_name)
 
-def get_quote(image_path):
+def get_quote(url):
     #reader = easyocr.Reader(['en'])
     #image = Image.open(image_1,mode = 'r')
     #image = Image.open(image_1)
 
     #image_2 = Image.open(requests.get(image_1, stream=True).raw)
 
-    image = Image.open(image_path)
+    #image = Image.open(image_path)
 
-    
-    
-    
-    
+    #url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+    with Image.open(requests.get(url, stream=True).raw) as image:
+        
+        image.save("cats.png")
+    #preds = predict(image)
     
     ##############
     pixel_values = feature_extractor(image, return_tensors="pt").pixel_values
@@ -59,7 +60,8 @@ def get_quote(image_path):
 #import gradio as gr
 title = "Image to text generation"
   
-demo = gr.Interface(fn=get_quote, inputs = "image", outputs=['text'],title = title, description = "Import an image file and get text from it" ,cache_examples=False, enable_queue=True).launch()
+demo = gr.Interface(fn=get_quote, inputs=gr.inputs.Image(type="pil"), outputs=['text'],title = title, description = "Import an image file and get text from it" ,cache_examples=False, enable_queue=True).launch()
+#inputs = "image"
 if __name__ == "__main__":
     
-    demo.launch()   
+    demo.launch(debug=True, cache_examples=True)