Update app.py

app.py

@@ -22,15 +22,14 @@ from funcs import (
 )
 from transformers import pipeline
 from diffusers import StableDiffusionXLPipeline
-import tensorflow as tf
 
-print(tf.__version__)
-print("GPU available:", len(tf.config.list_physical_devices('GPU')) > 0)
+print("PyTorch version:", torch.__version__)
+print("CUDA available:", torch.cuda.is_available())
 
 def is_tensor(x):
-    return tf.is_tensor(x)
+    return torch.is_tensor(x)
 
-os.environ['KERAS_BACKEND'] = 'tensorflow'
+os.environ['KERAS_BACKEND'] = 'pytorch'
 
 def download_model():
     REPO_ID = 'Doubiiu/DynamiCrafter_1024'