Update app.py

app.py

@@ -238,14 +238,13 @@ def inference(img,model_name):
       model_path = coco_weight
   else:
       model_path = conceptual_weight
-  model.load_state_dict(torch.load(model_path, map_location=CPU)) 
+  model.load_state_dict(torch.load(model_path, map_location=CPU), strict=False) 
   model = model.eval() 
   device = CUDA(0) if is_gpu else "cpu"
   model = model.to(device)
 
   use_beam_search = False 
-  image = io.imread(img.name)
-  pil_image = PIL.Image.fromarray(image)  
+  pil_image = PIL.Image.fromarray(img)  
   image = preprocess(pil_image).unsqueeze(0).to(device)
   with torch.no_grad():
       prefix = clip_model.encode_image(image).to(device, dtype=torch.float32)
@@ -263,11 +262,10 @@ article = "<p style='text-align: center'><a href='https://github.com/rmokady/CLI
 examples=[['water.jpeg',"COCO"]]
 gr.Interface(
     inference, 
-    [gr.inputs.Image(type="file", label="Input"),gr.inputs.Radio(choices=["COCO","Conceptual captions"], type="value", default="COCO", label="Model")], 
-    gr.outputs.Textbox(label="Output"),
+    inputs=[gr.Image(label="Input"),gr.Radio(choices=["COCO","Conceptual captions"], value="COCO", label="Model")], 
+    outputs=gr.Textbox(label="Output"),
     title=title,
     description=description,
     article=article,
-    enable_queue=True,
     examples=examples
     ).launch(debug=True)