Fix token

app.py

@@ -7,14 +7,8 @@ import re
 
 device = "cpu"
 is_peft = False
-model_id = "treadon/promt-fungineer-355M"
-# if is_peft:
-#     config = peft.PeftConfig.from_pretrained(model_id)
-#     model = transformers.AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, low_cpu_mem_usage=True)
-#     tokenizer = transformers.AutoTokenizer.from_pretrained(config.base_model_name_or_path)
-#     model = peft.PeftModel.from_pretrained(model, model_id)
-# else:
-auth_token = os.environ.get("hub_token") or True
+model_id = "treadon/prompt-fungineer-355M"
+auth_token = os.environ["HUB_TOKEN"]
 
 model = transformers.AutoModelForCausalLM.from_pretrained(model_id, low_cpu_mem_usage=True,use_auth_token=auth_token)
 tokenizer = transformers.AutoTokenizer.from_pretrained("gpt2")
@@ -63,16 +57,6 @@ def generate_text(prompt, extra=False, top_k=100, top_p=0.95, temperature=0.85,
     return samples
 
 
-# inputs = [
-#     gr.Textbox(lines=5, label="Base Prompt", placeholder="An astronaut in space", info="Enter a very simple prompt that will be fungineered into something exciting!"),
-#     gr.Checkbox(value=True, label="Extra Fungineer Imagination", info="If checked, the model will be allowed to go wild with its imagination."),
-#     gr.Slider( minimum=10, maximum=1000, value=100, label="Top K", info="Top K sampling"),
-#     gr.Slider( minimum=0.1, maximum=1, value=0.95, step=0.01, label="Top P", info="Top P sampling"),
-#     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"),
-# ]
-
-# iface = gr.Interface(fn=generate_text, inputs=inputs, outputs=["text","text","text","text"] )
-
 with gr.Blocks() as fungineer:
     with gr.Row():
         gr.Markdown("""# Midjourney / Dalle 2 / Stable Diffusion Prompt Generator