Update app.py

app.py

@@ -79,15 +79,11 @@ def inference(img, model):
     my_w = e4e_projection(aligned_face, "test.pt", device).unsqueeze(0)
     if model == 'JoJo':
         with torch.no_grad():
-            generator.eval()
         
-            #original_my_sample = original_generator(my_w, input_is_latent=True)
             my_sample = generatorjojo(my_w, input_is_latent=True)  
     else:
         with torch.no_grad():
-            generator.eval()
         
-            #original_my_sample = original_generator(my_w, input_is_latent=True)
             my_sample = generatordisney(my_w, input_is_latent=True)
     
     npimage = my_sample[0].permute(1, 2, 0).detach().numpy()