Update app.py

app.py

@@ -22,13 +22,11 @@ torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
 
 # model_repo_id = "John6666/wai-ani-nsfw-ponyxl-v8-sdxl"
-# model_repo_id = "John6666/wai-ani-nsfw-ponyxl-v140-sdxl"
-model_repo_id = "nncyberpunk/SDXL1.0_PonyDiffusionXL_6"
+model_repo_id = "John6666/wai-ani-nsfw-ponyxl-v140-sdxl"
 
 
 
-# pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype).to(device)
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, from_tf=True).to(device)
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024