MaawaKhalid commited on
Commit
5eebd83
·
verified ·
1 Parent(s): f2d9be5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import streamlit as st
2
- from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  # Define model and tokenizer
5
- model_name = 'openai-community/gpt2-large'
6
- model = AutoModelForCausalLM.from_pretrained(model_name)
7
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
8
 
9
  def generate_blogpost(topic):
10
  inputs = tokenizer.encode(topic, return_tensors='pt')
 
1
  import streamlit as st
2
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel
3
 
4
  # Define model and tokenizer
5
+ model_name = 'gpt2-large'
6
+ model = GPT2LMHeadModel.from_pretrained(model_name)
7
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
8
 
9
  def generate_blogpost(topic):
10
  inputs = tokenizer.encode(topic, return_tensors='pt')