Hugging Face's logo

Spaces:
Hayloo9838
/
dgapiv2
Runtime error

App Files Community
Hayloo9838 commited on
Commit
3f59d69
·
verified ·
1 Parent(s): f5eb9e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from io import BytesIO
10
  app = FastAPI()
11
 
12
  # load model once
13
- model_name = "nateraw/vit-base-patch16-224-in21k"
14
  model = AutoModelForImageClassification.from_pretrained(model_name)
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  transform = T.Compose([
 
10
  app = FastAPI()
11
 
12
  # load model once
13
+ model_name = "Falconsai/nsfw_image_detection"
14
  model = AutoModelForImageClassification.from_pretrained(model_name)
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  transform = T.Compose([