MunirAbobaker commited on
Commit
7dc943b
·
1 Parent(s): 83bd890

Add app file

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+
3
+ from transformers import pipeline
4
+
5
+ # Create a new FastAPI app instance
6
+ app = FastAPI()
7
+
8
+ # Initialize the text generation pipeline
9
+ # This function will be able to generate text
10
+ # given an input.
11
+ pipe = pipeline("text2text-generation",
12
+ model="google/flan-t5-small")
13
+
14
+ # Define a function to handle the GET request at `/generate`
15
+ # The generate() function is defined as a FastAPI route that takes a
16
+ # string parameter called text. The function generates text based on the # input using the pipeline() object, and returns a JSON response
17
+ # containing the generated text under the key "output"
18
+ @app.get("/generate")
19
+ def generate(text: str):
20
+ """
21
+ Using the text2text-generation pipeline from `transformers`, generate text
22
+ from the given input text. The model used is `google/flan-t5-small`, which
23
+ can be found [here](<https://huggingface.co/google/flan-t5-small>).
24
+ """
25
+ # Use the pipeline to generate text from the given input text
26
+ output = pipe(text)
27
+
28
+ # Return the generated text in a JSON response
29
+ return {"output": output[0]["generated_text"]}