Update app.py

app.py

@@ -3,11 +3,15 @@ import spaces
 import torch
 import random
 from diffusers import DiffusionPipeline, AutoPipelineForText2Image
+import os
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+model = os.getenv("MODEL_NAME", "Heartsync/NSFW-Uncensored")
+
+
 pipe = DiffusionPipeline.from_pretrained(
-    "Heartsync/NSFW-Uncensored",
+    model,
     torch_dtype=torch.float16
 )
 pipe.to(device)