Update app.py

- Update error with downloading everytime model when runs generation;
- Update look of 3DSkinView for mobile devices;

app.py

@@ -13,7 +13,6 @@ from io import BytesIO
 
 device = ("mps" if torch.backends.mps.is_available() else "cuda" if torch.cuda.is_available() else "cpu")
 
-
 class MSPipeline(DiffusionPipeline):
   def __init__(self, unet, scheduler):
     super().__init__()
@@ -36,6 +35,9 @@ class MSPipeline(DiffusionPipeline):
     return x
 
 
+pipe = MSPipeline.from_pretrained("WiNE-iNEFF/Mineskin-Diffusion-v1.0", use_safetensors=True).to(device)
+
+
 def clear_pix(x):
   datas = []
   for pixel in list(x.getdata()):
@@ -67,8 +69,6 @@ def show_3D(image, print_link = False):
 
 
 def generate(schedulers, inference_steps, images_num):
-  pipe = MSPipeline.from_pretrained("WiNE-iNEFF/Mineskin-Diffusion-v1.0", use_safetensors=True).to(device)
-
   if schedulers == "DDIMScheduler":
     pipe.scheduler = DDIMScheduler.from_pretrained("WiNE-iNEFF/Mineskin-Diffusion-v1.0", subfolder="scheduler")
   elif schedulers == "DDPMScheduler":
@@ -87,7 +87,7 @@ def generate(schedulers, inference_steps, images_num):
 
 
 def update_iframe(images):
-  iframe_html = "<div style='display: grid; grid-template-columns: repeat(2, 1fr); gap: 10px'>"
+  iframe_html = "<div style='display: grid; gap: 10px'>" #grid-template-columns: repeat(2, 1fr); 
   for img in images:
     iframe_url = show_3D(clear_pix(img), print_link=True)
     iframe_html += f"<iframe style='min-width: 100%;' src='{iframe_url}'></iframe>"