Spaces:

Bunpheng
/

foodieproject

Runtime error

Bunpheng commited on Sep 6

Commit

68973dc

•

1 Parent(s): 4d384cd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ app = FastAPI()
 pipe_flan = pipeline("text2text-generation", model="mistralai/Mistral-7B-v0.3")
 @app.get("/mistral")
-def t5(input):
     output = pipe_flan(input)
     return {"output": output[0]["generated_text"]}

 pipe_flan = pipeline("text2text-generation", model="mistralai/Mistral-7B-v0.3")
 @app.get("/mistral")
+def mistral(input: str):
     output = pipe_flan(input)
     return {"output": output[0]["generated_text"]}