Staticaliza commited on
Commit
64b2e38
·
verified ·
1 Parent(s): 6d6dac5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ footer {
41
  }
42
  '''
43
 
44
- nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
45
 
46
  repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
47
  repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
@@ -133,7 +133,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
133
 
134
  print(image_paths)
135
 
136
- nsfw_prediction = nsfw_classifier(image_paths[0])
137
 
138
  print(nsfw_prediction)
139
 
 
41
  }
42
  '''
43
 
44
+ repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
45
 
46
  repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
47
  repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 
133
 
134
  print(image_paths)
135
 
136
+ nsfw_prediction = repo_nsfw_classifier(image_paths[0])
137
 
138
  print(nsfw_prediction)
139