| import gradio as gr |
| from transformers import pipeline |
| import torch |
|
|
| |
| def load_model(): |
| try: |
| return pipeline("text2text-generation", |
| model="prithivida/parrot_paraphraser_on_T5", |
| device="cuda" if torch.cuda.is_available() else "cpu") |
| except: |
| return None |
|
|
| parrot_pipeline = load_model() |
|
|
| def paraphrase_text(text): |
| if not text.strip(): |
| return "⚠️ Please enter text" |
| |
| if parrot_pipeline is None: |
| return "❌ Model unavailable. Try again later." |
| |
| try: |
| outputs = parrot_pipeline( |
| text, |
| num_return_sequences=3, |
| num_beams=5, |
| max_length=128 |
| ) |
| return "\n\n".join([f"{i+1}. {res['generated_text']}" |
| for i, res in enumerate(outputs)]) |
| except Exception as e: |
| return f"⚠️ Error: {str(e)}" |