Update app.py

app.py

@@ -12,6 +12,21 @@ import yt_dlp
 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'
+
+# Cacher le nom du repo
+
+python_path = hf_hub_download(repo_id='PierreHanna/TextRetrieval', repo_type="space", filename='models.py',
+                                use_auth_token='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')
+
+#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'])
+print(python_path)
+sys.path.append(os.environ['PRIVATE_DIR'])
+from models import *
+
 def download_audio_(link):
   with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(title)s.mp3'}) as video:
     info_dict = video.extract_info(link, download = True)