Spaces:

efrenam
/

testing

Running

efrenam commited on Jan 11

Commit

da2b21e

verified ·

1 Parent(s): 3f9754c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import pipeline
 import gradio as gr
 # Initialize the text generation pipeline
-generator = pipeline("text-generation", model="gpt2")
 def generate_response(prompt):
     """Generate text based on the user's prompt."""

 import gradio as gr
 # Initialize the text generation pipeline
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 def generate_response(prompt):
     """Generate text based on the user's prompt."""