bmas10 commited on
Commit
78c666c
·
verified ·
1 Parent(s): b81a54b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -16,6 +16,8 @@ messages = [
16
  pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
17
  pipe(messages)
18
 
 
 
19
  def chat(input_text, history=[]):
20
  history.append(input_text)
21
  prompt = "\n".join(history) + "\nAI:" # Simple conversational format
 
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)
23
  prompt = "\n".join(history) + "\nAI:" # Simple conversational format