fish 10 months ago
parent
commit
ecb702a171
2 changed files with 5 additions and 2 deletions
  1. 2 1
      demo/demo1.py
  2. 3 1
      demo/demo2.py

+ 2 - 1
demo/demo1.py

@@ -7,5 +7,6 @@ def to_black(image):
 
 # GUI      http://127.0.0.1:7860/
 interface = gr.Interface(fn=to_black, inputs="image", outputs="image",  examples=[["test.png"]])
-interface.launch()
 
+if __name__=='__main__':
+    interface.launch(inbrowser=True, server_name='0.0.0.0')

+ 3 - 1
demo/demo2.py

@@ -15,4 +15,6 @@ def predict(inp):
   return {labels[i]: float(prediction[i]) for i in range(1000)}
 inputs = gr.inputs.Image()
 outputs = gr.outputs.Label(num_top_classes=3)
-gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()
+
+if __name__=='__main__':
+    gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()