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 |
|