File size: 729 Bytes
b935e3d f404d24 ed5d53c b935e3d ed5d53c b935e3d ed0ae5f dee82dd f404d24 ed0ae5f f404d24 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import gradio as gr
from transformers import pipeline
import torch
# Check if a GPU is available
device = 0 if torch.cuda.is_available() else -1
print("Using GPU" if device == 0 else "Using CPU")
# Load the model on the GPU if available
model = pipeline("text-generation", model="gpt2", device=device)
def generate_text(prompt):
return model(prompt, max_length=50)[0]["generated_text"]
interface = gr.Interface(
fn=generate_text,
inputs=gr.Textbox(label="Enter your prompt here"),
outputs=gr.Textbox(label="Generated Text"),
title="AI Text Generator",
description="This app generates text based on your input prompt. Try it out!",
theme="dark"
)
if __name__ == "__main__":
interface.launch()
|