GenAICoder commited on
Commit
ada32e6
·
verified ·
1 Parent(s): 9a09cff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -14,7 +14,13 @@ st.title("DeepSeek Chatbot")
14
  prompt = st.text_input("Enter your message:")
15
 
16
  if st.button("Run"):
17
- inputs = tokenizer.encode(prompt, return_tensors="pt")
18
- outputs = model.generate(inputs)
19
- response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 
 
 
 
 
20
  st.text_area("Long Text Box", response, height=1000)
 
14
  prompt = st.text_input("Enter your message:")
15
 
16
  if st.button("Run"):
17
+ messages =[]
18
+ messages.append({"role": "user", "content": prompt})
19
+ response= model.chat.completions.create(
20
+ model="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
21
+ messages=messages,
22
+ temperature=0.6,
23
+ max_tokens=1024,
24
+ top_p=0.95
25
+ )
26
  st.text_area("Long Text Box", response, height=1000)