Update app.py

app.py

@@ -16,11 +16,11 @@ login(token=a )
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-#model_repo_id = "stabilityai/sdxl-turbo" # "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
+model_repo_id = "stabilityai/sdxl-turbo" # "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
 #model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 #model_repo_id = "cagliostrolab/animagine-xl-4.0"
 #model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
-model_repo_id ="black-forest-labs/FLUX.1-dev"
+#model_repo_id ="black-forest-labs/FLUX.1-dev"
 
 
 
@@ -70,7 +70,7 @@ def infer(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    generator = torch.Generator("cpu").manual_seed(seed)
+    generator = torch.Generator().manual_seed(seed)
 
     image = pipe(
         prompt=prompt,
@@ -156,7 +156,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
-                    value=3.5,  # Replace with defaults that work for your model
+                    value=0.0,  # Replace with defaults that work for your model
                 )
 
                 num_inference_steps = gr.Slider(
@@ -164,7 +164,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=50,
                     step=1,
-                    value=50,  # Replace with defaults that work for your model
+                    value=3,  # Replace with defaults that work for your model
                 )
 
         gr.Examples(examples=examples, inputs=[prompt])