Update app.py

app.py

@@ -11,15 +11,16 @@ from huggingface_hub import hf_hub_download
 import uuid
 import traceback
 
+
 embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
 embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
 # NO GPU
 #os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
-#import tensorflow 
+import tensorflow 
 #tensorflow.config.threading.set_intra_op_parallelism_threads(8)
 #tensorflow.config.threading.set_inter_op_parallelism_threads(8)
-
+print(tf.config.list_physical_devices('GPU'))
 #
 python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],
                                 use_auth_token=os.environ['TOKEN'])