Update app.py

app.py

@@ -3,14 +3,12 @@ from transformers.utils.hub import move_cache
 import torch
 from torch import autocast
 from diffusers import StableDiffusionPipeline, DDIMScheduler
-import os
 
 move_cache()
 
 model_path = "slymnyldrm/dreambooth_usecase_weights"
-commandline_args = os.environ.get('COMMANDLINE_ARGS', "--skip-torch-cuda-test --no-half")
 
-pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float32)
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
 g_cuda = torch.Generator()
@@ -32,7 +30,7 @@ with gr.Blocks() as demo:
             negative_prompt = gr.Textbox(label="Negative Prompt", value="")
             run = gr.Button(value="Generate")
             with gr.Row():
-                num_samples = gr.Number(label="Number of Samples", value=4)
+                num_samples = gr.Number(label="Number of Samples", value=1)
                 guidance_scale = gr.Number(label="Guidance Scale", value=7.5)
             with gr.Row():
                 height = gr.Number(label="Height", value=512)