Upload app.py with huggingface_hub

app.py

@@ -4,7 +4,9 @@ import time
 import uuid
 import hashlib
 import subprocess
+import huggingface as hh
 from flask import Flask, request, send_file
+
 MAX_SECONDS = 100
 server_me = 'https://zaisheng-testapi.hf.space'
 
@@ -95,10 +97,18 @@ def putvideo():
     h = request.headers
     key = h['Key']
     file = h['File']
+    repo_id_video = h['RV']
+    repo_id_prompt = h['RP']
+    tt = h['tt']
     if not get_key_sha256(key) == match_sha256:
         return 'err'
     f = request.files['file']
-    f.save(os.path.join(video_folder, file+'.mp4'))
+    local_file_video = os.path.join(video_folder, file+'.mp4')
+    local_file_prompt = os.path.join(prompt_folder, file)
+    f.save(local_file_video)
+    hh.upload_file(path_or_fileobj=local_file_video, path_in_repo=file+'.mp4', repo_id=repo_id_video, repo_type='dataset', token=tt)
+    hh.upload_file(path_or_fileobj=local_file_prompt, path_in_repo=file, repo_id=repo_id_prompt, repo_type='dataset', token=tt)
+    
     return "ok"