Rockramsri commited on
Commit
544a0eb
1 Parent(s): 7a6a53f
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,4 +9,6 @@ install("llama-cpp-python")
9
  from llama_cpp import Llama
10
  prompt = st.chat_input("Say something")
11
  if prompt:
12
- st.write(f"User has sent the following prompt: {prompt}")
 
 
 
9
  from llama_cpp import Llama
10
  prompt = st.chat_input("Say something")
11
  if prompt:
12
+ llm = Llama(model_path="Meta-Llama-3.1-8B-Instruct-Q4_K_M.gguf")
13
+ r=llm(prompt, max_tokens=1000)
14
+ st.write(f"User has sent the following prompt: {prompt} with response: {r["choices"][0]["text"]} ")