Update app.py

app.py

@@ -53,7 +53,11 @@ def download_image(url):
     response = requests.get(url)
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
 
-device = "cpu" #"cuda" if torch.cuda.is_available() else "cpu"
+#device = "cpu" #"cuda" if torch.cuda.is_available() else "cpu"
+
+device = torch.device(“cuda:0” if torch.cuda.is_available() else “cpu”)
+print(“The model will be running on”, device, “device”)
+# Convert model parameters and buffers to CPU or Cuda
 
 model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
@@ -118,7 +122,7 @@ def predict(radio, dict, word_mask, prompt=""):
         
         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
-        prompt = labels.to(torch.device(device))
+        prompt = prompt.to(torch.device(device))
         #---------
         
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
@@ -149,7 +153,7 @@ def predict(radio, dict, word_mask, prompt=""):
         
         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
-        prompt = labels.to(torch.device(device))
+        prompt = prompt.to(torch.device(device))
         
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         filename = f"{uuid.uuid4()}.png"