updates

main.py

@@ -13,37 +13,38 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler, Eule
 app = FastAPI() 
 
 @app.get("/generate")
-def generate_image(prompt, inference_steps, model):
+def generate_image(prompt, model):
     torch.cuda.empty_cache()
 
     pipeline = StableDiffusionPipeline.from_pretrained(str(model), torch_dtype=torch.float16)
     pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
     pipeline = pipeline.to("cuda")
  
-    image = pipeline(prompt, num_inference_steps=int(inference_steps), height=512, width=512).images[0]
+    image = pipeline(prompt, num_inference_steps=50, height=512, width=512).images[0]
 
     filename = str(uuid.uuid4()) + ".jpg"
     image.save(filename)
     
     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)
+                "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)
 
     subprocess.check_output(['./truepic-sign', 'init', 'file-system', '--api-key', os.environ.get("api_key")])
-    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions-file', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
+#    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions-file', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
+    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', assertion, '--output', (os.getcwd() + '/static/' + filename)])
     
     return {"response": filename}
 

static/index.html

@@ -57,15 +57,6 @@
             id="text-gen-input"
           ></textarea>
 
-          <label>Inference Steps: </label>
-          <select id="inference_steps">
-            <option disabled selected value>Select</option>
-            <option value="50">50</option>
-            <option value="20">20</option>
-            <option value="10">10</option>
-            <option value="5">5</option>
-          </select>
-
           <label>Model: </label>
           <select id="model">
             <option disabled selected value>Select</option>

static/script.js

@@ -1,7 +1,6 @@
 
 const textGenForm = document.querySelector('.text-gen-form');
 const textGenInput = document.getElementById('text-gen-input');
-const inferenceSteps = document.getElementById('inference_steps');
 const model = document.getElementById('model');
 const textGenSubmit = document.getElementById('text-gen-submit');
 const spinner = document.getElementById('spinner');
@@ -14,21 +13,21 @@ const certificateOutput = document.getElementById('certificate-output');
 const verificationDetails = document.querySelector('.verification-details');
 
 
-[textGenInput, inferenceSteps, model].forEach(item => {
+[textGenInput, model].forEach(item => {
   item.addEventListener('change', async (event) => {
     setButtonStatus()
   });
 })
 
 const setButtonStatus = () => {
-  if (textGenInput.value && inferenceSteps.value && model.value) 
+  if (textGenInput.value && model.value) 
     textGenSubmit.classList.add("active");
   else 
     textGenSubmit.classList.remove("active");
 }
 
-const generateImage = async (text, inference_steps, model) => {
-    const inferResponse = await fetch(`generate?prompt=${text}&inference_steps=${inference_steps}&model=${model}`);
+const generateImage = async (text, model) => {
+    const inferResponse = await fetch(`generate?prompt=${text}&model=${model}`);
     const inferJson = await inferResponse.json();
     
     return inferJson.response;
@@ -45,7 +44,7 @@ textGenForm.addEventListener('submit', async (event) => {
       if (document.getElementById('result')) document.getElementById('result').remove();
       spinner.style.display = "block";
 
-      const resp = await generateImage(textGenInput.value, inferenceSteps.value, model.value);
+      const resp = await generateImage(textGenInput.value, model.value);
       const path = "/" + resp;
       
       var resultsContainer = document.getElementById('image-container');