djrana commited on
Commit
a9d5136
1 Parent(s): 5c272b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ text_generator = pipeline(
17
  )
18
 
19
  # Load tokenizer and model for image generation
20
- tokenizer = AutoTokenizer.from_pretrained("stablediffusionapi/juggernaut-xl-v8")
21
- model = AutoModelForCausalLM.from_pretrained("stablediffusionapi/juggernaut-xl-v8")
22
 
23
  # Function to generate text based on input prompt
24
  def generate_text(prompt):
 
17
  )
18
 
19
  # Load tokenizer and model for image generation
20
+ tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
21
+ model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-2.7B")
22
 
23
  # Function to generate text based on input prompt
24
  def generate_text(prompt):