silterra commited on
Commit
6d9b318
1 Parent(s): 7fc9185

Tweaks for updated Gradio

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.py +2 -2
Dockerfile CHANGED
@@ -30,4 +30,4 @@ EXPOSE $PORT
30
  # Run streamlit app under conda environment
31
  # CMD ["sh", "-c", "streamlit run --server.port=$PORT --server.address=0.0.0.0 app.py"]
32
 
33
- CMD ["python", "main.py"]
 
30
  # Run streamlit app under conda environment
31
  # CMD ["sh", "-c", "streamlit run --server.port=$PORT --server.address=0.0.0.0 app.py"]
32
 
33
+ CMD ["python3", "main.py"]
main.py CHANGED
@@ -19,8 +19,8 @@ def predict(inp):
19
  def run():
20
  demo = gr.Interface(
21
  fn=predict,
22
- inputs=gr.inputs.Image(type="pil"),
23
- outputs=gr.outputs.Label(num_top_classes=3),
24
  )
25
 
26
  demo.launch(server_name="0.0.0.0", server_port=7860)
 
19
  def run():
20
  demo = gr.Interface(
21
  fn=predict,
22
+ inputs=gr.Image(type="pil"),
23
+ outputs=gr.Label(num_top_classes=3),
24
  )
25
 
26
  demo.launch(server_name="0.0.0.0", server_port=7860)