Spaces:
Sleeping
Sleeping
HF_TOKEN Updated
Browse files
app.py
CHANGED
|
@@ -593,6 +593,8 @@ import gradio as gr
|
|
| 593 |
from gradio import FileData
|
| 594 |
import time
|
| 595 |
import spaces
|
|
|
|
|
|
|
| 596 |
ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
|
| 597 |
model = MllamaForConditionalGeneration.from_pretrained(ckpt,
|
| 598 |
torch_dtype=torch.bfloat16).to("cuda")
|
|
|
|
| 593 |
from gradio import FileData
|
| 594 |
import time
|
| 595 |
import spaces
|
| 596 |
+
from huggingface_hub import login
|
| 597 |
+
login(token=os.getenv("HF_TOKEN"))
|
| 598 |
ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
|
| 599 |
model = MllamaForConditionalGeneration.from_pretrained(ckpt,
|
| 600 |
torch_dtype=torch.bfloat16).to("cuda")
|