Update app.py

app.py

@@ -166,12 +166,12 @@ def inference(img, model):
     aligned_face = align_face('out.jpg')
         
     my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)
-    if model == 'JoJo':
+    if model == 'Joker':
         with torch.no_grad():
-            my_sample = generatorjojo(my_w, input_is_latent=True)  
-    elif model == 'Disney':
+            my_sample = generatorjoker(my_w, input_is_latent=True)  
+    elif model == 'Voldemort':
         with torch.no_grad():
-            my_sample = generatordisney(my_w, input_is_latent=True)
+            my_sample = generatorvoldemort(my_w, input_is_latent=True)
     # elif model == 'Jinx':
     #     with torch.no_grad():
     #         my_sample = generatorjinx(my_w, input_is_latent=True)