added main method

app.py

@@ -202,31 +202,33 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     return output, input_image, fig
 
 
-demo = gr.Interface(
-    fn=predict_and_analyze,
-    inputs=[gr.Dropdown(["regnet", "efficientnet"], 
-                        value="efficientnet",
-                        label="Model Selection",
-                        show_label=True), 
-            gr.Dropdown(["45", "61", "75"], 
-                        value="61",
-                        label="Number of Velocity Channels",
-                        show_label=True), 
-            gr.Dropdown(["600"], 
-                        value="600",
-                        label="Image Dimensions",
-                        show_label=True), 
-            gr.File(label="Input Data", show_label=True)],
-    outputs=[gr.Textbox(lines=1, label="Prediction", show_label=True), 
-             gr.Image(label="Input Image", show_label=True), 
-            #  gr.Image(label="Activation 1", show_label=True), 
-            #  gr.Image(label="Actication 2", show_label=True)],
-             gr.Plot(label="Activations", show_label=True) 
-            #  gr.Plot(label="Actication 2", show_label=True)],
-            ],
-    title="Kinematic Planet Detector"
-)
-demo.launch()
+if __name__ == "__main__":
+
+    demo = gr.Interface(
+        fn=predict_and_analyze,
+        inputs=[gr.Dropdown(["regnet", "efficientnet"], 
+                            value="efficientnet",
+                            label="Model Selection",
+                            show_label=True), 
+                gr.Dropdown(["45", "61", "75"], 
+                            value="61",
+                            label="Number of Velocity Channels",
+                            show_label=True), 
+                gr.Dropdown(["600"], 
+                            value="600",
+                            label="Image Dimensions",
+                            show_label=True), 
+                gr.File(label="Input Data", show_label=True)],
+        outputs=[gr.Textbox(lines=1, label="Prediction", show_label=True), 
+                gr.Image(label="Input Image", show_label=True), 
+                #  gr.Image(label="Activation 1", show_label=True), 
+                #  gr.Image(label="Actication 2", show_label=True)],
+                gr.Plot(label="Activations", show_label=True) 
+                #  gr.Plot(label="Actication 2", show_label=True)],
+                ],
+        title="Kinematic Planet Detector"
+    )
+    demo.launch()