Update app.py

app.py

@@ -45,6 +45,8 @@ config.blip_num_beams = 64
 ci = Interrogator(config)
 
 def inference(image, mode, best_max_flavors):
+    image = image.to(config.device)
+
     image = image.convert('RGB')
     if mode == 'best':
         
@@ -155,8 +157,8 @@ with gr.Blocks(css=css) as block:
             share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
 
         examples=[['eg1.jpeg', "best",4], ['eg2.jpeg',"fast",4]]
-        #ex = gr.Examples(examples=examples, fn=inference, inputs=[input_image, mode_input, flavor_input], outputs=[output_text, share_button, community_icon, loading_icon], cache_examples=True, run_on_click=True)
-        #ex.dataset.headers = [""]
+        ex = gr.Examples(examples=examples, fn=inference, inputs=[input_image, mode_input, flavor_input], outputs=[output_text, share_button, community_icon, loading_icon], cache_examples=True, run_on_click=True)
+        ex.dataset.headers = [""]
         
         gr.HTML(article)