Spaces:

tchans123
/

gpt2-deployment

Sleeping

tchans123 commited on Sep 24, 2024

Commit

12c1e66

1 Parent(s): 257a345

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import pipeline
-# Load the FLAN-T5 XXL model
-generator = pipeline("text2text-generation", model="google/flan-t5-xxl")
 def generate_text(prompt):
     # Generate text using FLAN-T5 XXL

 import gradio as gr
 from transformers import pipeline
+generator = pipeline("text2text-generation", model="google/flan-t5-base")  # or flan-t5-large
 def generate_text(prompt):
     # Generate text using FLAN-T5 XXL