MANI2003 commited on
Commit
1a7c53e
·
verified ·
1 Parent(s): 460a6ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load text-to-text generation pipeline
5
- generator = pipeline("text2text-generation", model="google/flan-t5-base")
6
 
7
  def generate_text(input_text):
8
  result = generator(input_text, max_length=100, clean_up_tokenization_spaces=True)
 
2
  from transformers import pipeline
3
 
4
  # Load text-to-text generation pipeline
5
+ generator = pipeline("text2text-generation", model="Qwen/Qwen2.5-7B-Instruct")
6
 
7
  def generate_text(input_text):
8
  result = generator(input_text, max_length=100, clean_up_tokenization_spaces=True)