Spaces:
Paused
Paused
Upload main.py
Browse files
main.py
CHANGED
@@ -20,9 +20,7 @@ pipe = pipeline("text2text-generation", model="google/flan-t5-large")
|
|
20 |
@app.get("/generate")
|
21 |
def generate(text: str):
|
22 |
"""
|
23 |
-
|
24 |
-
from the given input text. The model used is `google/flan-t5-small`, which
|
25 |
-
can be found [here](https://huggingface.co/google/flan-t5-small).
|
26 |
"""
|
27 |
output = pipe(text)
|
28 |
return {"output": output[0]["generated_text"]}
|
|
|
20 |
@app.get("/generate")
|
21 |
def generate(text: str):
|
22 |
"""
|
23 |
+
deployed flan-t5-large model as backend
|
|
|
|
|
24 |
"""
|
25 |
output = pipe(text)
|
26 |
return {"output": output[0]["generated_text"]}
|