LPX55 commited on
Commit
433b282
·
verified ·
1 Parent(s): 4ebf62d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,12 +1,13 @@
1
  import gradio as gr
2
- from transformers import AutoBackbone, AutoModelForImageClassification, pipeline, AutoImageProcessor, Swinv2ForImageClassification
 
3
  from torchvision import transforms
4
 
5
  # model = AutoModelForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
6
  # image_processor = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
7
-
8
- image_processor = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
9
- model = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy", out_indices=(1,))
10
 
11
  clf = pipeline(model=model, task="image-classification", image_processor=image_processor)
12
 
 
1
  import gradio as gr
2
+ # from transformers import AutoBackbone, AutoModelForImageClassification, AutoImageProcessor, Swinv2ForImageClassification
3
+ from transformers import pipeline, AutoImageProcessor, Swinv2ForImageClassification, Swinv2Model
4
  from torchvision import transforms
5
 
6
  # model = AutoModelForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
7
  # image_processor = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
8
+ image_processor = AutoImageProcessor.from_pretrained("haywoodsloan/ai-image-detector-deploy")
9
+ # image_processor = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
10
+ model = Swinv2ForImageClassification.from_pretrained("haywoodsloan/ai-image-detector-deploy")
11
 
12
  clf = pipeline(model=model, task="image-classification", image_processor=image_processor)
13