bmas10 commited on
Commit
5075f8a
·
verified ·
1 Parent(s): b64b77b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -13,10 +13,13 @@ from transformers import pipeline
13
  messages = [
14
  {"role": "user", "content": "Who are you?"},
15
  ]
16
- pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
 
 
 
17
  pipe(messages)
18
 
19
- model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
20
 
21
  def chat(input_text, history=[]):
22
  history.append(input_text)
 
13
  messages = [
14
  {"role": "user", "content": "Who are you?"},
15
  ]
16
+
17
+ model_name = "dbmas10/DeepSeek-Llama-8-GGUF"
18
+
19
+ pipe = pipeline("text-generation", model=model_name)
20
  pipe(messages)
21
 
22
+
23
 
24
  def chat(input_text, history=[]):
25
  history.append(input_text)