IcedSundae commited on
Commit
4303cdd
·
verified ·
1 Parent(s): 76a64a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,8 +3,8 @@ from langchain.llms import CTransformers
3
 
4
  # Load model
5
  llm = CTransformers(
6
- model="TheBloke/OpenHermes-2.5-Mistral-7B-GGUF",
7
- model_type="mistral",
8
  config={"gpu_layers": 0} # Change to GPU layers if using paid plan
9
  )
10
 
@@ -13,5 +13,5 @@ def chat(message):
13
  return response
14
 
15
  # Create UI
16
- iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="OpenHermes Chatbot")
17
  iface.launch()
 
3
 
4
  # Load model
5
  llm = CTransformers(
6
+ model="IcedSundae/Krog-v1",
7
+ model_type="llama",
8
  config={"gpu_layers": 0} # Change to GPU layers if using paid plan
9
  )
10
 
 
13
  return response
14
 
15
  # Create UI
16
+ iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="Krog")
17
  iface.launch()