schizoooo commited on
Commit
39d99af
·
verified ·
1 Parent(s): 675ab4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -20,8 +20,15 @@ cloudinary.config(
20
  )
21
 
22
  MODEL_NAME = "prithivMLmods/Realistic-Gender-Classification"
23
- model = AutoModelForImageClassification.from_pretrained(MODEL_NAME)
24
- processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
 
 
 
 
 
 
 
25
 
26
  @app.route('/classify', methods=['POST'])
27
  def classify():
 
20
  )
21
 
22
  MODEL_NAME = "prithivMLmods/Realistic-Gender-Classification"
23
+
24
+ model = AutoModelForImageClassification.from_pretrained(
25
+ MODEL_NAME,
26
+ trust_remote_code=True
27
+ )
28
+ processor = AutoImageProcessor.from_pretrained(
29
+ MODEL_NAME,
30
+ trust_remote_code=True
31
+ )
32
 
33
  @app.route('/classify', methods=['POST'])
34
  def classify():