Spaces:
Runtime error
Runtime error
AnishKumbhar
commited on
Commit
•
1999fab
1
Parent(s):
c6d7b15
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline
|
|
7 |
app = FastAPI(docs_url="/")
|
8 |
|
9 |
@app.get("/generate")
|
10 |
-
def
|
11 |
"""
|
12 |
Using the text2text-generation pipeline from `transformers`, generate text
|
13 |
from the given input text. The model used is `google/flan-t5-small`, which
|
|
|
7 |
app = FastAPI(docs_url="/")
|
8 |
|
9 |
@app.get("/generate")
|
10 |
+
def generate_endpoint(breed: str):
|
11 |
"""
|
12 |
Using the text2text-generation pipeline from `transformers`, generate text
|
13 |
from the given input text. The model used is `google/flan-t5-small`, which
|