Neupane9Sujal commited on
Commit
57ed3b9
1 Parent(s): 9fb5ac0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -10,6 +10,11 @@ def get_llm():
10
  llm = AutoModelForCausalLM.from_pretrained("TheBloke/CodeLlama-7B-GGUF", model_type="llama")
11
  return llm
12
 
 
 
 
 
 
13
  @app.get("/generate")
14
  def generate(text:str) ->dict:
15
  llm = get_llm()
 
10
  llm = AutoModelForCausalLM.from_pretrained("TheBloke/CodeLlama-7B-GGUF", model_type="llama")
11
  return llm
12
 
13
+ @app.get("/")
14
+ def index():
15
+ return {"Hello, World!"}
16
+
17
+
18
  @app.get("/generate")
19
  def generate(text:str) ->dict:
20
  llm = get_llm()