Staticaliza commited on
Commit
6d64ddf
·
verified ·
1 Parent(s): 75c2377

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
121
 
122
  print(image_paths)
123
 
124
- nsfw_prediction = repo_nsfw_classifier(**processor(images=Image.open(image_paths[0]), return_tensors="pt")).logits
125
 
126
  print(nsfw_prediction)
127
 
 
121
 
122
  print(image_paths)
123
 
124
+ nsfw_prediction = repo_nsfw_classifier(**processor_nsfw_classifier(images=Image.open(image_paths[0]), return_tensors="pt")).logits
125
 
126
  print(nsfw_prediction)
127