| import gradio as gr |
| from transformers import pipeline |
|
|
| models = ["gpt2", "microsoft/phi-2","meta-llama/Llama-2-7b-chat-hf" ] |
|
|
| def generate_text(prompt, model, max_length, temperature): |
| pipe = pipeline("text-generation", model=model,trust_remote_code = True) |
| try: |
| generated_text = pipe(prompt, max_length=max_length, temperature=temperature)[0]["generated_text"] |
| return generated_text |
| except Exception as e: |
| return f"An error occurred: {str(e)}" |
|
|
| gr.Interface( |
| generate_text, |
| [ |
| gr.Textbox(label="Enter your Prompt"), |
| gr.Dropdown(models, label="choose your Model" ), |
| gr.Slider(minimum=50, maximum=650, step=50), |
| gr.Slider(minimum=0.1, maximum=1, step=0.1), |
| ], |
| outputs="text" |
| ).launch() |
|
|
