changes

.gitignore

@@ -3,3 +3,4 @@ model.pth
 scratch.ipynb
 sparsify.ipynb
 Untitled-1.py
+testing_pipelines.ipynb

GIANNIS

@@ -1 +1 @@
-Subproject commit ac14f85ffa5e31a729781061172ecc4042644bc2
+Subproject commit 7a0f21e34b5fc9d30ada99349d6be6fed9301845

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
-from diffusers import DDPMPipeline, DDPMScheduler
+from diffusers import PNDMPipeline, PNDMScheduler
 import torch
 import random
 import PIL
 import os
 
-scheduler = DDPMScheduler(num_train_timesteps=200,beta_schedule="linear")
-pipeline = DDPMPipeline.from_pretrained("uripper/GIANNIS", revision="ONNX", scheduler=scheduler)
-num_imgs = len(os.listdir("GIANNIS\stored_images"))
-imgs = os.listdir("GIANNIS\stored_images")
+scheduler = PNDMScheduler(num_train_timesteps=1000,beta_schedule="linear")
+pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
+# num_imgs = len(os.listdir("\stored_images"))
+# imgs = os.listdir("\stored_images")
 
 
 def generate_img():
@@ -27,13 +27,13 @@ def generate_img():
         pass
     images= pipeline().images
     im = images[0]
-    im.save(f"GIANNIS\stored_images\{filename}.png")
+    # im.save(f"GIANNIS\stored_images\{filename}.png")
     return im
 
-def show_random_image(num_imgs = num_imgs, imgs = imgs):
-    ran_img = random.randint(1, num_imgs)-1
-    display_img = "GIANNIS\\stored_images\\"+imgs[ran_img] 
-    return PIL.Image.open(display_img)
+# def show_random_image(num_imgs = num_imgs, imgs = imgs):
+#     ran_img = random.randint(1, num_imgs)-1
+#     display_img = "GIANNIS\\stored_images\\"+imgs[ran_img] 
+#     return PIL.Image.open(display_img)
 
 # for i in range(100):
 #     generate_img()