Update app.py

app.py

@@ -74,21 +74,40 @@ def infer(lora_1_id, lora_1_sfts, lora_2_id, lora_2_sfts, prompt, negative_promp
 
     pipe.to("cuda")
 
-    pipe.load_lora_weights(
-        lora_1_id,
-        weight_name = lora_1_sfts,        
-        low_cpu_mem_usage = True,
-        use_auth_token = True
-    )
+    if lora_1_sfts == "NO SAFETENSORS FILE": 
+        pipe.load_lora_weights(
+            lora_1_id,     
+            low_cpu_mem_usage = True,
+            use_auth_token = True
+        )
+
+    else:
+        pipe.load_lora_weights(
+            lora_1_id,
+            weight_name = lora_1_sfts,        
+            low_cpu_mem_usage = True,
+            use_auth_token = True
+        )
+
+    
 
     pipe.fuse_lora(lora_1_scale)
 
-    pipe.load_lora_weights(
-        lora_2_id,
-        weight_name = lora_2_sfts,        
-        low_cpu_mem_usage = True,
-        use_auth_token = True
-    )
+    if lora_2_sfts == "NO SAFETENSORS FILE": 
+        pipe.load_lora_weights(
+            lora_2_id,     
+            low_cpu_mem_usage = True,
+            use_auth_token = True
+        )
+
+    else:
+        pipe.load_lora_weights(
+            lora_2_id,
+            weight_name = lora_2_sfts,        
+            low_cpu_mem_usage = True,
+            use_auth_token = True
+        )
+
 
     pipe.fuse_lora(lora_2_scale)