Update app.py

app.py

@@ -12,7 +12,7 @@ comic_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 # Stable Diffusion Model
 model_id = "stabilityai/sd-turbo"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
 pipe.to("cpu")
 
 # Function to generate comic strip
@@ -26,7 +26,7 @@ def generate_comic(user_prompt, num_panels, art_choice):
     comic_images = []
     for panel in comic_panels:
         prompt = f"{panel}, {art_choice} style, bold outlines, vibrant colors"
-        image = pipe(prompt).images[0]
+        image = pipe(prompt, num_inference_steps=30, do_sample=True, temperature=0.7).images[0]
         comic_images.append(image)
     
     # Step 3: Create a Grid Layout for Comic Strip