Update main.py

main.py

@@ -34,27 +34,27 @@ def generate_image(prompt, inference_steps, model):
     
     print(filename)
     
-    # Data to be written
-    # assertion = {
-    #            "assertions": [
-    #                {
-    #                "label": "com.truepic.custom.ai",
-    #                "data": {
-    #                    "model_name": model,
-    #                    "model_version": "1.0",
-    #                    "prompt": prompt
-    #                }
-    #                }
-    #            ]
-    #            }
+    Data to be written
+    assertion = {
+               "assertions": [
+                   {
+                   "label": "com.truepic.custom.ai",
+                   "data": {
+                       "model_name": model,
+                       "model_version": "1.0",
+                       "prompt": prompt
+                   }
+                   }
+               ]
+               }
      
-    # json_object = json.dumps(assertion, indent=4)
-    # with open("assertion.json", "w") as outfile:
-    #    outfile.write(json_object)
+    json_object = json.dumps(assertion, indent=4)
+    with open("assertion.json", "w") as outfile:
+       outfile.write(json_object)
 
     subprocess.check_output(['./truepic-sign', 'init', 'file-system', '--api-key', os.environ.get("api_key")])
-    # subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
-    subprocess.check_output(['./truepic-sign', 'sign', filename, '--output', (os.getcwd() + '/static/' + filename)])
+    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
+    # subprocess.check_output(['./truepic-sign', 'sign', filename, '--output', (os.getcwd() + '/static/' + filename)])
 
     return {"response": filename}