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")
|