minor changes to running the model

app.py

@@ -1,7 +1,7 @@
 from diffusers import StableDiffusionPipeline
 import torch
-work_around_for_hugging_face_gradio_sdk_bug = "/blob/main/rem_3k.ckpt"
-model_url = "https://huggingface.co/waifu-research-department/Rem" + work_around_for_hugging_face_gradio_sdk_bug
+file_name = "/blob/main/rem_3k.ckpt"
+model_url = "https://huggingface.co/waifu-research-department/Rem" + file_name
 pipeline = StableDiffusionPipeline.from_single_file(
     model_url,
     torch_dtype=torch.float16,
@@ -48,22 +48,19 @@ image # your image is saved in this PIL variable
 ```
 """
 
-
-log = "GPU available"
-try :
+try : 
   pipeline.to("cuda")
-except : 
+except: 
   log = "no GPU available"
 
 
 def text2img(positive_prompt,negative_prompt):
-    global log
-    if log == "no GPU available":
-        image = None
-        return log,image
-    else : 
-        image = pipeline(positive_prompt,negative_prompt=negative_prompt).images[0]
-        log = {"postive_prompt":positive_prompt,"negative_prompt":negative_prompt}
-        return log,image
+    try : 
+      image = pipeline(positive_prompt,negative_prompt=negative_prompt).images[0]
+      log = {"postive_prompt":positive_prompt,"negative_prompt":negative_prompt}
+    except Exception as e:
+      log = f"ERROR: {e}"
+      image = None
+    return log,image
 
 gr.Interface(text2img,["text","text"],["text","image"],examples=[["rem","3D"]],description=description).launch()