Spaces:
Runtime error
Runtime error
Jacob Silterra commited on
Commit ·
6d9b318
1
Parent(s): 7fc9185
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)
|