Spaces:
Sleeping
Sleeping
try double quote
Browse files- Dockerfile +1 -1
- main.py +0 -1
Dockerfile
CHANGED
|
@@ -24,4 +24,4 @@ COPY --chown=user . $HOME/app
|
|
| 24 |
|
| 25 |
|
| 26 |
|
| 27 |
-
CMD ["python",
|
|
|
|
| 24 |
|
| 25 |
|
| 26 |
|
| 27 |
+
CMD ["python", "main.py"]
|
main.py
CHANGED
|
@@ -5,7 +5,6 @@ import gradio as gr
|
|
| 5 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 6 |
|
| 7 |
|
| 8 |
-
# @app.get("/infer_t5")
|
| 9 |
def pred(input):
|
| 10 |
output = pipe_flan(input)
|
| 11 |
return output[0]["generated_text"]
|
|
|
|
| 5 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 6 |
|
| 7 |
|
|
|
|
| 8 |
def pred(input):
|
| 9 |
output = pipe_flan(input)
|
| 10 |
return output[0]["generated_text"]
|