Chanlefe commited on
Commit
28a20f8
·
verified ·
1 Parent(s): f64fb33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -4,7 +4,11 @@ from transformers import AutoProcessor, AutoModelForImageClassification
4
  import gradio as gr
5
 
6
  # Load model and processor
7
- model_name = "your-username/your-siglip2-meme-classifier" # Or local path
 
 
 
 
8
  model = AutoModelForImageClassification.from_pretrained(model_name)
9
  processor = AutoProcessor.from_pretrained(model_name)
10
 
 
4
  import gradio as gr
5
 
6
  # Load model and processor
7
+ from transformers import AutoModelForImageClassification, AutoProcessor
8
+
9
+ model = AutoModelForImageClassification.from_pretrained("model")
10
+ processor = AutoProcessor.from_pretrained("model")
11
+
12
  model = AutoModelForImageClassification.from_pretrained(model_name)
13
  processor = AutoProcessor.from_pretrained(model_name)
14