Hugging Face's logo

Spaces:
ahmadmac
/
Pretrain-GPT
Sleeping

App Files Community
ahmadmac commited on
Commit
f7b8f0c
·
verified ·
1 Parent(s): 2d24340

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ st.set_page_config(page_title="GPT-2 Text Generator", layout="centered")
6
 
7
  #tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrained-GPT2")
8
  #model = AutoModelForCausalLM.from_pretrained("ahmadmac/Pretrained-GPT2")
9
- tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrain-CSV-GPT2")
10
- model = AutoModelForCausalLM.from_pretrained("ahmadmac/results")
11
  #google_api_key=
12
  import google.generativeai as genai
13
  GOOGLE_API_KEY=os.environ["google_api_key"]
 
6
 
7
  #tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrained-GPT2")
8
  #model = AutoModelForCausalLM.from_pretrained("ahmadmac/Pretrained-GPT2")
9
+ tokenizer = AutoTokenizer.from_pretrained("ahmadmac/DistillGPT2-CSV")
10
+ model = AutoModelForCausalLM.from_pretrained("ahmadmac/DistillGPT2-CSV")
11
  #google_api_key=
12
  import google.generativeai as genai
13
  GOOGLE_API_KEY=os.environ["google_api_key"]