Upload app.py with huggingface_hub

app.py

@@ -24,13 +24,15 @@ def put():
     key = d['key']
     if not get_key_sha256(key) == match_sha256:
         return 'err'
-    machine_type = d['machine_type']
+    machine_type = d.get('machine_type', 'default')
     job = d['job']
     sha256 = get_key_sha256(job)
     folder = os.path.join(data_folder, 'new', machine_type)
     doing_folder = os.path.join(data_folder, 'doing', machine_type)
+    done_folder = os.path.join(data_folder, 'done', machine_type)
     os.makedirs(folder, exist_ok=True)
     os.makedirs(doing_folder, exist_ok=True)
+    os.makedirs(done_folder, exist_ok=True)
     file = os.path.join(folder, sha256)
     open(file, 'w').write(job)
     return 'ok'
@@ -59,9 +61,9 @@ def get():
         os.remove(source_file)
     except:
         print('deleteerr')
-    return txt
+    return json.dumps({'sha256': file, 'job': txt})
 
-@app.route("/done", methods = ['GET'])
+@app.route("/done", methods = ['POST'])
 def done():
     d = request.json
     key = d['key']
@@ -69,6 +71,7 @@ def done():
         return 'err'
     sha256 = d['sha256']
     machine_type = d['machine_type']
+    output = d['output']
     file = os.path.join(data_folder, 'doing', machine_type, sha256)
     try:
         os.remove(file)
@@ -88,7 +91,6 @@ def get_folder_list(folder, full=False):
                 dd[f] = open(os.path.join(folder, machine_type, f)).read()
             d[machine_type] = dd
         else:
-            
             d[machine_type] = fs
     return d