Update app.py

app.py

@@ -58,7 +58,7 @@ def inference(img):
     pred_class_names = [imagenet_id_to_classname[str(i.item())] for i in pred_classes[0]]
     return "Top 5 predicted labels: %s" % ", ".join(pred_class_names)
     
-inputs = gr.inputs.Image(type='filepath')
+inputs = gr.inputs.Image(type='pil')
 outputs = gr.outputs.Textbox(label="Output")
 
 title = "Omnivore"
@@ -68,4 +68,4 @@ description = "Gradio demo for Revisiting Weakly Supervised Pre-Training of Visu
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2201.08371' target='_blank'>Revisiting Weakly Supervised Pre-Training of Visual Perception Models</a> | <a href='https://github.com/facebookresearch/SWAG' target='_blank'>Github Repo</a></p>"
 
 
-gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=[['library.jpg']]).launch(enable_queue=True)
+gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=[['library.jpg']]).launch(enable_queue=True,cache_examples=True)