Aditya12kumar commited on
Commit
fc8c977
·
verified ·
1 Parent(s): 8b8d155

Update fast-api_test.py

Browse files
Files changed (1) hide show
  1. fast-api_test.py +11 -1
fast-api_test.py CHANGED
@@ -1,6 +1,16 @@
1
  from fastapi import FastAPI
 
 
2
  app = FastAPI()
3
 
 
 
 
4
  @app.get("/")
5
  def root():
6
- return {"message": "This is Quote Generation API test"}
 
 
 
 
 
 
1
  from fastapi import FastAPI
2
+ from .QuoteGenerator import QuoteGenerator
3
+
4
  app = FastAPI()
5
 
6
+ quote_generator = QuoteGenerator()
7
+ quote_generator.load_generator()
8
+
9
  @app.get("/")
10
  def root():
11
+ return {"message": "This is Quote Generation API test"}
12
+
13
+ @app.post("/generate_quote")
14
+ def generate_quote_for_user(prompt):
15
+ generated_quote = quote_generator.generate_quote(prompt)
16
+ return generated_quote