Update app.py

app.py

@@ -34,10 +34,18 @@ def run_inference(prompt, stable_diffusion_model, num_inference_steps, guidance_
 
     # view it also as 3d model or not output
     if model_3d:
-        return os.path.join(f"output_minecraft_skins/{filename}"), os.path.join(f"output_minecraft_skins/{filename}_3d_model.glb")
+        image_path = os.path.join(f"output_minecraft_skins/{filename}")
+        image = Image.open(image_path)
+        image = image.resize((64, 64), resample=Image.NEAREST)  # Use nearest neighbor interpolation
+        image.save(image_path)
+        return image_path, os.path.join(f"output_minecraft_skins/{filename}_3d_model.glb")
     else:
-        return os.path.join(f"output_minecraft_skins/{filename}"), None
-
+        image_path = os.path.join(f"output_minecraft_skins/{filename}")
+        image = Image.open(image_path)
+        image = image.resize((64, 64), resample=Image.NEAREST)  # Use nearest neighbor interpolation
+        image.save(image_path)
+        return image_path, None
+        
 
 # Define Gradio UI components
 prompt = gr.Textbox(label="Your Prompt", info="What the Minecraft Skin should look like")