22

app.py

@@ -59,9 +59,10 @@ pipe_lora = QwenImagePipeline.from_pretrained(
 )
 
 pipe_imagen = QwenImagePipeline.from_pretrained(
-    "Qwen/Qwen-Image", 
-    torch_dtype=DTYPE
-).to("cuda")
+    "Qwen/Qwen-Image",
+    torch_dtype=DTYPE,
+    device="cuda"
+)
 
 
 @spaces.GPU

static/header.html

@@ -6,6 +6,6 @@
         <p>
             Generate LoRA from a few images.
             <br>
-            If you like my work, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊 -->
+            If you like my work, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊 
     </div>
 </div>