minor edits

app.py

@@ -83,14 +83,14 @@ pipes = {
 }
 # Define Gradio inputs and outputs
 inputs = [
-    gr.inputs.Image(type="pil", label="Image"),
-    gr.inputs.Textbox(label="Candidate Labels (comma-separated)"),
-    gr.inputs.Radio(
+    gr.Image(type="pil", label="Image", width=300, height=300),
+    gr.Textbox(label="Candidate Labels (comma-separated)"),
+    gr.Radio(
         choices=["ViT/B-16", "ViT/L-14@336px-base-text"],
         label="Model",
         value="ViT/B-16",
     ),
-    gr.inputs.Textbox(label="Prompt Template", placeholder="Optional prompt template as prefix",
+    gr.Textbox(label="Prompt Template", placeholder="Optional prompt template as prefix",
                value=""),
 ]
 outputs = gr.Label(label="Predicted Scores")
@@ -129,7 +129,8 @@ iface = gr.Interface(shot,
             description="""<p>Demo for UniMed CLIP, a family of strong Medical Contrastive VLMs trained on UniMed-dataset. For more information about our project, refer to our paper and github repository. <br>
             Paper: <a href='https://arxiv.org/abs/2412.10372'>https://arxiv.org/abs/2412.10372</a> <br>
             Github: <a href='https://github.com/mbzuai-oryx/UniMed-CLIP'>https://github.com/mbzuai-oryx/UniMed-CLIP</a> <br><br>
-            <b>[DEMO USAGE]</b> To begin with the demo, provide a picture (either upload manually, or select from the given examples) and class labels. Optionally you can also add template as an prefix to the class labels. <br> </p>""",
+            <b>[DEMO USAGE]</b> To begin with the demo, provide a picture (either upload manually, or select from the given examples) and class labels. Optionally you can also add template as an prefix to the class labels. <br> <b>[NOTE]</b> This demo is running on CPU and thus the response time might be a bit slower. Running it on a machine with a GPU will result in much faster predictions. </p>""",
+
             title="Zero-shot Medical Image Classification with UniMed-CLIP")
 
-iface.launch()
+iface.launch(allowed_paths=["/home/user/app/docs/sample_images"])