Antigravity Bot commited on
Commit
8e27a9e
·
1 Parent(s): 3e5704d

Upgrade to DeepSeek-R1-Distill-Qwen-7B

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # 1. Carrega o modelo (Isso estava funcionando)
5
- pipe = pipeline("text-generation", model="Qwen/Qwen2.5-Coder-1.5B-Instruct", device_map="cpu")
6
 
7
  def predict(message, history):
8
  """
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # 1. Carrega o modelo (DeepSeek R1 para raciocínio)
5
+ pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-7B", device_map="cpu", torch_dtype="auto")
6
 
7
  def predict(message, history):
8
  """