add hugginface login
Browse files
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()
|