Antigravity Bot commited on
Commit
34306bf
·
1 Parent(s): 11747ae

Tmp switch to 0.5B for debug

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # 1. Carrega o modelo (Isso já estava funcionando)
5
- pipe = pipeline("text-generation", model="Qwen/Qwen2.5-Coder-3B-Instruct", device_map="cpu")
6
 
7
  def predict(message, history):
8
  """
@@ -42,7 +42,7 @@ demo = gr.ChatInterface(
42
  fn=predict,
43
  type="messages", # <--- ESTA LINHA CORRIGE O AVISO E MUDA O FORMATO DO HISTÓRICO
44
  title="Qwen Agent Node",
45
- description="Bot rodando Qwen 2.5 Coder 3B via CPU no Hugging Face."
46
  )
47
 
48
  if __name__ == "__main__":
 
2
  from transformers import pipeline
3
 
4
  # 1. Carrega o modelo (Isso já estava funcionando)
5
+ pipe = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct", device_map="cpu")
6
 
7
  def predict(message, history):
8
  """
 
42
  fn=predict,
43
  type="messages", # <--- ESTA LINHA CORRIGE O AVISO E MUDA O FORMATO DO HISTÓRICO
44
  title="Qwen Agent Node",
45
+ description="Bot rodando Qwen 2.5 0.5B Debug via CPU no Hugging Face."
46
  )
47
 
48
  if __name__ == "__main__":