Update app.py

app.py

@@ -2,8 +2,9 @@ from diffusers import DiffusionPipeline
 import matplotlib.pyplot as plt
 import torch
 import gradio as gr
+device = "cuda" if torch.cuda.is_available() else "cpu"
 pipeline = DiffusionPipeline.from_pretrained("anusha-bhambore/live-eventful")
-pipeline = pipeline.to("cuda")
+pipeline = pipeline.to(device)
 def generate_image_interface(prompt, negative_prompt,num_inference_steps=50, weight=640):
   params = {'prompt': prompt, 'num_inference_steps': num_inference_steps, 'num_images_per_prompt':2, 'height':int(1.2*weight),
             'weight': weight, 'negative_prompt': negative_prompt}