Rockramsri commited on
Commit
2c502cc
1 Parent(s): a0d4907
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -12,7 +12,11 @@ llm = Llama(
12
 
13
  @app.get("/")
14
  def home():
15
- return {"message":"Hello World"}
 
 
 
 
16
 
17
  # Define a function to handle the GET request at `/generate`
18
 
 
12
 
13
  @app.get("/")
14
  def home():
15
+ print("helloe here")
16
+ output= llm("What is the difference btw RAG and Fine tunning", max_tokens=1000)
17
+ print(output["choices"][0]["text"])
18
+ ## return the generate text in Json reposne
19
+ return {"output":output["choices"][0]["text"]}
20
 
21
  # Define a function to handle the GET request at `/generate`
22