itmerk commited on
Commit
203d55d
1 Parent(s): 72a4dbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -1,19 +1,24 @@
1
  from fastapi import FastAPI
2
  from transformers import pipeline
3
 
4
- app = FastAPI()
 
5
 
 
6
  pipe = pipeline("text2text-generation", model="google/flan-t5-small")
7
 
 
8
  @app.get("/")
9
  def home():
10
- return {'message':'Hello World'}
 
 
 
11
 
12
- # Define a function to handle the GET request at '/generate'
13
  @app.get("/generate")
14
  def generate(text:str):
15
- # Use the pipeline to generate text from given input text
16
- output = pipe(text)
17
 
18
- # Return the generate text in JSON request
19
  return {"output":output[0]['generated_text']}
 
1
  from fastapi import FastAPI
2
  from transformers import pipeline
3
 
4
+ ## create a new FASTAPI app instance
5
+ app=FastAPI()
6
 
7
+ # Initialize the text generation pipeline
8
  pipe = pipeline("text2text-generation", model="google/flan-t5-small")
9
 
10
+
11
  @app.get("/")
12
  def home():
13
+ return {"message":"Hello World"}
14
+
15
+ # Define a function to handle the GET request at `/generate`
16
+
17
 
 
18
  @app.get("/generate")
19
  def generate(text:str):
20
+ ## use the pipeline to generate text from given input text
21
+ output=pipe(text)
22
 
23
+ ## return the generate text in Json reposne
24
  return {"output":output[0]['generated_text']}