Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
-import torch
+import tensorflow as tf
 from semdiffusers import SemanticEditPipeline
 
-device='cuda'
+device = 'gpu' if tf.config.list_physical_devices('GPU') else 'cpu'
 
 pipe = SemanticEditPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
 ).to(device)
 
 def infer(prompt, seed):
-    gen = torch.Generator(device=device)
-    gen.manual_seed(seed)
+    gen = tf.random.Generator.from_seed(seed)
     
     out = pipe(
         prompt=prompt,