Update app.py
Browse files
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(**
|
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 |
|