Add key

.gitignore

@@ -0,0 +1 @@
+venv

app.py

@@ -2,6 +2,7 @@
 import gradio as gr
 #import peft
 import transformers
+import os
 
 device = "cpu"
 is_peft = False
@@ -12,7 +13,9 @@ model_id = "treadon/promt-fungineer-355M"
 #     tokenizer = transformers.AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 #     model = peft.PeftModel.from_pretrained(model, model_id)
 # else:
-model = transformers.AutoModelForCausalLM.from_pretrained(model_id, low_cpu_mem_usage=True)
+auth_token = os.environ.get("hub_token") or True
+
+model = transformers.AutoModelForCausalLM.from_pretrained(model_id, low_cpu_mem_usage=True,use_auth_token=auth_token)
 tokenizer = transformers.AutoTokenizer.from_pretrained("gpt2")
 
 def generate_text(prompt):