Spaces:

wasmdashai
/

meta-llama-Llama-3.2-11B-Vision-Instruct

Sleeping

wasmdashai commited on Nov 9, 2024

Commit

bd303e3

verified ·

1 Parent(s): 356762a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,10 @@ import gradio as gr
 from gradio import FileData
 import time
 import spaces
 ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
-model = MllamaForConditionalGeneration.from_pretrained(ckpt,
     torch_dtype=torch.bfloat16).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt)

 from gradio import FileData
 import time
 import spaces
+import os
+token=os.environ.get("key_")
 ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
+model = MllamaForConditionalGeneration.from_pretrained(ckpt,token=token,
     torch_dtype=torch.bfloat16).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt)