Update app.py

app.py

@@ -392,22 +392,21 @@ def run_inversion(self, dict, pcs, epochs, weight_decay,lr):
 def file_upload(file):
     device="cuda"
     weights = torch.load(file.name).to(device)
-    net = "model_"+str(uuid.uuid4())[:4]+".pt"
-    torch.save(weights, net)
         
     #pad to 10000 Principal components to keep everything consistent
     pcs = net.shape[1]
     padding =  torch.zeros((1,10000-pcs)).to(device)
     weights = torch.cat((weights, padding), 1)
+    net = "model_"+str(uuid.uuid4())[:4]+".pt"
+    torch.save(weights, net)
        
-        
     image = prompt = "sks person"
     negative_prompt = "low quality, blurry, unfinished, nudity, weapon"
     seed = 5
     cfg = 3.0
     steps = 25
     image = inference(net, prompt, negative_prompt, cfg, steps, seed)
-    return net,net,image
+    return net,image