frankrobotics commited on
Commit
633c1bf
Β·
verified Β·
1 Parent(s): dca0cf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -2,8 +2,15 @@ import gradio as gr
2
 
3
  import torch
4
  # from transformers import pipeline
 
5
 
6
- model_id = "deepseek-ai/DeepSeek-R1"
 
 
 
 
 
 
7
  messages = [
8
  {"role": "user", "content": "Who are you?"},
9
  ]
 
2
 
3
  import torch
4
  # from transformers import pipeline
5
+ import huggingface_hub
6
 
7
+ import os
8
+
9
+ hf_token = os.getenv('HF_TOKEN')
10
+
11
+ huggingface_hub.login(token=hf_token)
12
+
13
+ model_id = "meta-llama/Llama-3.2-3B-Instruct"
14
  messages = [
15
  {"role": "user", "content": "Who are you?"},
16
  ]