Update app.py

app.py

@@ -16,38 +16,30 @@ os.chdir("Stable_Diffusion_Finetuned_Minecraft_Skin_Generator")
 
 
 def generate(
-    system_prompt,
     prompt,
-    max_new_tokens,
-    repetition_penalty,
-    temperature,
-    top_p,
-    top_k,
+    stable_diffusion_model,
+    num_inference_steps,
+    guidance_scale,
+    num_images_per_prompt,
+    model_precision_type,
     seed
 ):
     
-    input_text = f"{system_prompt}, {prompt}"
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
-
-    if seed == 0:
-        seed = random.randint(1, 100000)
-        torch.manual_seed(seed)
+    if verbose:
+        verbose_opt = '--verbose'
     else:
-        torch.manual_seed(seed)
+        verbose_opt = ''
         
-    outputs = model.generate(
-        input_ids,
-        max_new_tokens=max_new_tokens,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        temperature=temperature,
-        top_p=top_p,
-        top_k=top_k,
-    )
-
-    better_prompt = tokenizer.decode(outputs[0])
-    better_prompt = better_prompt.replace("<pad>", "").replace("</s>", "")
-    return better_prompt
+    if stable_diffusion_model == '2':
+        sd_model = "minecraft-skins"
+    else:
+        sd_model = "minecraft-skins-sdxl"
+    
+    command = f"Python_Scripts/{sd_model}.py '{prompt}' {num_inference_steps} {guidance_scale} {num_images_per_prompt} {model_precision_type} {output_image_name} {verbose_opt}"
+    
+    subprocess.run(["python", "$command"])
+    
+    return os.path.join(f"output_minecraft_skins/{output_image_name}")
 
 
 prompt = gr.Textbox(label="Prompt", interactive=True)
@@ -64,6 +56,10 @@ model_precision_type = gr.Dropdown(["fp16", "fp32"], value="fp16" interactive=Tr
 
 seed = gr.Number(value=42, interactive=True, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
 
+output_image_name = gr.Textbox(label="Name of Generated Skin Output", interactive=Trie, value="output.png")
+
+verbose = gr.Checkbox(label="Verbose Output", interactive=True, value=False, info="Produce verbose output while running")
+
 examples = [
     [
         "A man in a purple suit wearing a tophat.",
@@ -78,7 +74,7 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, stable_diffusion_model, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, seed],
+    inputs=[prompt, stable_diffusion_model, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, output_image_name, verbose, seed],
     outputs=gr.Image(label="Generated Minecraft Skin"),
     title="Stable Diffusion Finetuned Minecraft Skin Generator",
     description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",