Fix parameters and ENV values

app.py

@@ -7,8 +7,13 @@ import re
 
 device = "cpu"
 is_peft = False
-model_id = "treadon/prompt-fungineer-355M"
-auth_token = os.environ["HUB_TOKEN"]
+model_id = os.environ.get("MODEL_ID") or "treadon/prompt-fungineer-355M"
+auth_token = os.environ.get("HUB_TOKEN") or True
+
+print(f"Using model {model_id}.")
+
+if auth_token != True:
+    print("Using auth token.")
 
 model = transformers.AutoModelForCausalLM.from_pretrained(model_id, low_cpu_mem_usage=True,use_auth_token=auth_token)
 tokenizer = transformers.AutoTokenizer.from_pretrained("gpt2")
@@ -72,9 +77,9 @@ Just enter a basic prompt and the fungineering model will use its wildest imagin
                 temperature = gr.Slider( minimum=0.1, maximum=1.2, value=0.85, step=0.01, label="Temperature", info="Temperature sampling.  Higher values will make the model more creative")
 
             with gr.Accordion("Advanced Output Settings", open=False):
-                gr.Checkbox(value=True, label="Enhancers", info="Add image meta information such as lens type, shuffter speed, camera model, etc.")
-                gr.Checkbox(value=False, label="Inpsiration", info="Include inspirational photographers that are known for this type of photography.  Sometimes random people will appear here, needs more training.")
-                gr.Checkbox(value=False, label="Negative Prompt", info="Include a negative prompt, more often used in Stable Diffusion.  If you're a Stable Diffusion user, chances are you already have a better negative prompt you like to use.")
+                enh = gr.Checkbox(value=True, label="Enhancers", info="Add image meta information such as lens type, shuffter speed, camera model, etc.")
+                insp = gr.Checkbox(value=False, label="Inpsiration", info="Include inspirational photographers that are known for this type of photography.  Sometimes random people will appear here, needs more training.")
+                neg = gr.Checkbox(value=False, label="Negative Prompt", info="Include a negative prompt, more often used in Stable Diffusion.  If you're a Stable Diffusion user, chances are you already have a better negative prompt you like to use.")
 
         with gr.Column():
             outputs = [
@@ -84,7 +89,7 @@ Just enter a basic prompt and the fungineering model will use its wildest imagin
                 gr.Textbox(lines=5, label="Fungineered Text 4"),
             ]
 
-    inputs = [base_prompt, extra, top_k, top_p, temperature]
+    inputs = [base_prompt, extra, top_k, top_p, temperature, enh, insp, neg]
 
 
     submit = gr.Button(label="Fungineer",variant="primary")