made1570 commited on
Commit
c0ba723
·
verified ·
1 Parent(s): c742ff6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
  # Load processor and model
10
- model_name = "adarsh3601/my_gemma_pt3" # Change to your model path
11
  processor = AutoProcessor.from_pretrained(model_name)
12
  model = AutoModelForImageTextToText.from_pretrained(model_name).to(device)
13
 
 
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
  # Load processor and model
10
+ model_name = "adarsh3601/my_gemma3_pt" # Change to your model path
11
  processor = AutoProcessor.from_pretrained(model_name)
12
  model = AutoModelForImageTextToText.from_pretrained(model_name).to(device)
13