Spaces:
Sleeping
Sleeping
Tweaks for updated Gradio
Browse files- Dockerfile +1 -1
- 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 ["
|
|
|
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.
|
23 |
-
outputs=gr.
|
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)
|