Update app.py

app.py

@@ -288,7 +288,7 @@ def generate_30(
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.txt'
+        filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
         print('-- generating image --')
         #with torch.no_grad():
@@ -341,7 +341,7 @@ def generate_60(
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.txt'
+        filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
         print('-- generating image --')
         #with torch.no_grad():
@@ -394,7 +394,7 @@ def generate_90(
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename= f'rv_IP_{timestamp}.txt'
+        filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
         print('-- generating image --')
         #with torch.no_grad():