pratikshahp commited on
Commit
39357c0
·
verified ·
1 Parent(s): 06cfd6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -5,13 +5,15 @@ from dotenv import load_dotenv
5
  import os
6
  load_dotenv()
7
 
 
 
8
 
9
- hf_token=os.getenv("HF_TOKEN")
10
 
11
  # Load the model and tokenizer
12
- model_name = "openai-community/gpt2"
13
- tokenizer = AutoTokenizer.from_pretrained(model_name)
14
- model = AutoModelForCausalLM.from_pretrained(model_name,token=hf_token)
15
 
16
  # Function to generate blog content
17
  def generate_blog(topic, keywords):
 
5
  import os
6
  load_dotenv()
7
 
8
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
9
+ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
10
 
11
+ #hf_token=os.getenv("HF_TOKEN")
12
 
13
  # Load the model and tokenizer
14
+ #model_name = "openai-community/gpt2"
15
+ #tokenizer = AutoTokenizer.from_pretrained(model_name)
16
+ #model = AutoModelForCausalLM.from_pretrained(model_name,token=hf_token)
17
 
18
  # Function to generate blog content
19
  def generate_blog(topic, keywords):