Aashi commited on
Commit
cd7160b
1 Parent(s): cb8cb23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,11 +2,10 @@ import gradio as gr
2
  import requests
3
  import torch
4
  from PIL import Image
 
5
  from transformers import MllamaForConditionalGeneration, AutoProcessor
6
 
7
- from huggingface_hub import login
8
- login("token_spaces", add_to_git_credential=True)
9
-
10
  # Load the Llama 3.2 Vision Model
11
  def load_llama_model():
12
  model_id = "meta-llama/Llama-3.2-11B-Vision"
 
2
  import requests
3
  import torch
4
  from PIL import Image
5
+ import spaces
6
  from transformers import MllamaForConditionalGeneration, AutoProcessor
7
 
8
+ @spaces.GPU
 
 
9
  # Load the Llama 3.2 Vision Model
10
  def load_llama_model():
11
  model_id = "meta-llama/Llama-3.2-11B-Vision"