Update app.py
Browse files
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("
|
21 |
-
model = AutoModelForCausalLM.from_pretrained("
|
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):
|