llama-3.2-vision-11B

Running on Zero

rphrp1985 commited on Sep 30, 2024

Commit

20712dc

verified ·

1 Parent(s): 132e348

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,6 +7,8 @@ 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")

 from gradio import FileData
 import time
 import spaces
+from PIL import Image
+from io import BytesIO
 ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
 model = MllamaForConditionalGeneration.from_pretrained(ckpt,
     torch_dtype=torch.bfloat16).to("cuda")