Adding image preview for URLs

app.py

@@ -63,10 +63,10 @@ def classify_image_url(learn, url_text):
         img = im.to_thumb(256,256)
         #resize_images(dest, max_size=400, dest=dest)
         os.remove(dest)
-        return classify_image(learn, img)
+        return classify_image(learn, img),img
     except:
         # in case there is any error, invalid URL or invalid image, etc., not sure how Gradio will handle a runtime exception so catching it to be safe
-        return { categories[0]: 0.0, categories[1]: 0.0 }
+        return { categories[0]: 0.0, categories[1]: 0.0 },None
 
 
 def classify_image_url_debug(url_text):
@@ -94,8 +94,11 @@ import gradio as gr
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("Rudimentary age predictor.  No refinement, just a hacked together experiment to try multiple output classes with fast.ai.  Many ways the accuracy could be improved.  See: https://www.kaggle.com/code/zachwormgoor/age-predictor \r\n")
-    gr.Markdown("Predict age from uploaded image or from provide URL to image file:")
+    gr.Markdown("Rudimentary age predictor.  No refinement, just a hacked together experiment to try multiple output classes with fast.ai.  Many ways the accuracy could be improved.  ")
+    gr.Markdown("See: https://www.kaggle.com/code/zachwormgoor/age-predictor  ")
+    gr.Markdown("  ")
+    gr.Markdown("Predict age from uploaded image or from provide URL to image file:  ")
+    gr.Markdown("  ")
     with gr.Tabs():
         with gr.TabItem("Image - resnet18"):
             with gr.Row():
@@ -124,9 +127,9 @@ with demo:
             text_d_button = gr.Button("Predict - debug")
 
     image_r_button.click(classify_image_resnet,       inputs=img_r_input,  outputs=img_r_output)
-    url_r_button.click(  classify_image_url_resnet,   inputs=text_r_input, outputs=text_r_output)
+    url_r_button.click(  classify_image_url_resnet,   inputs=text_r_input, outputs=(text_r_output,gr.Image()))
     image_c_button.click(classify_image_convnext,     inputs=img_c_input,  outputs=img_c_output)
-    text_c_button.click( classify_image_url_convnext, inputs=text_c_input, outputs=text_d_output)
+    text_c_button.click( classify_image_url_convnext, inputs=text_c_input, outputs=(text_c_output, gr.Image()))
     text_d_button.click( classify_image_url_debug,    inputs=text_d_input, outputs=text_d_output)
  
 demo.launch()