Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline, set_seed
 
-generator = pipeline('text-generation', model='flax-community/miniLM-L6-h384-uncased', device=0)
+generator = pipeline('text-generation', model='google/palm-2-large-uncased', device=0)
 
 def generate_text(prompt, length=50, temperature=0.7, seed=42):
     set_seed(seed)
@@ -17,7 +17,7 @@ logging.basicConfig(level=logging.INFO)
 def generate_text(prompt, length=50, temperature=0.7, seed=42):
     try:
         set_seed(seed)
-        generator = pipeline('text-generation', model='flax-community/miniLM-L6-h384-uncased', device=0)
+        generator = pipeline('text-generation', model='google/palm-2-large-uncased', device=0)
         output = generator(prompt, max_length=length, do_sample=True, temperature=temperature)
         return output[0]['generated_text']
     except Exception as e:
@@ -31,9 +31,10 @@ length_slider = gr.inputs.Slider(minimum=10, maximum=200, default=50, label="Len
 seed_input = gr.inputs.Number(default=42, label="Seed")
 
 title = "Generative AI"
-description = "Use MiniLM to generate text based on a prompt."
+description = "Use PaLM 2 to generate text based on a prompt."
 examples = [["The quick brown fox", 50, 0.7, 42]]
 
 iface = gr.Interface(fn=generate_text, inputs=[inputs, length_slider, temperature_slider, seed_input], outputs=outputs, title=title, description=description, examples=examples)
 
 iface.launch(inbrowser=True, share=True)
+