saurluca commited on
Commit
851f27f
·
verified ·
1 Parent(s): ac65d96

add hugginface login

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -2,6 +2,10 @@ import gradio as gr
2
  import spaces
3
  import torch
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
 
5
 
6
  # model_name = "saurluca/Apertus-8B-Instruct-2509-bnb-4bit"
7
  model_name = "swiss-ai/Apertus-8B-Instruct-2509"
@@ -38,4 +42,4 @@ def greet(prompt: str):
38
  return output
39
 
40
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
41
- demo.launch()
 
2
  import spaces
3
  import torch
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
+ from huggingface_hub import login
6
+
7
+ # Log in to Hugging Face with your token from secrets
8
+ login(token="hf_token")
9
 
10
  # model_name = "saurluca/Apertus-8B-Instruct-2509-bnb-4bit"
11
  model_name = "swiss-ai/Apertus-8B-Instruct-2509"
 
42
  return output
43
 
44
  demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
45
+ demo.launch()