Kryko7 commited on
Commit
023ea2b
1 Parent(s): 32f5623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ app = FastAPI()
7
  pipe = pipeline("text-generation", model="google/flan-t5-small")
8
 
9
 
10
- @app.get("/generate")
11
  def generate(query: str):
12
  output = pipe(query, max_length=50, do_sample=True, top_k=50)
13
  return {"output": output[0]["generated_text"], "query": query, "model": "google/flan-t5-small"}
 
7
  pipe = pipeline("text-generation", model="google/flan-t5-small")
8
 
9
 
10
+ @app.get("/")
11
  def generate(query: str):
12
  output = pipe(query, max_length=50, do_sample=True, top_k=50)
13
  return {"output": output[0]["generated_text"], "query": query, "model": "google/flan-t5-small"}