Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,9 @@ from transformers import pipeline
|
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
@app.get("/infer_t5")
|
| 13 |
def t5(input):
|
|
|
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 11 |
+
#Inference Augmentation: Load a dataset like SQuAD for question-answering;
|
| 12 |
+
#to generate responses, testing zero-shot capabilities
|
| 13 |
+
|
| 14 |
|
| 15 |
@app.get("/infer_t5")
|
| 16 |
def t5(input):
|