Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|