pip install transformers import gradio as gr from transformers import GPT2LMHeadModel, GPT2Tokenizer # Load the GPT-2 model and tokenizer model_name = "gpt2" model = GPT2LMHeadModel.from_pretrained(model_name) tokenizer = GPT2Tokenizer.from_pretrained(model_name) # Define the sentence completion function def complete_sentence(sentence): input_ids = tokenizer.encode(sentence, return_tensors="pt") output = model.generate(input_ids, max_length=50, num_return_sequences=1) completed_sentence = tokenizer.decode(output[0], skip_special_tokens=True) return completed_sentence # Create the Gradio interface iface = gr.Interface( fn=complete_sentence, inputs="text", outputs="text", title="Sentence Completion", description="Enter a sentence to complete", example="I love to" ) # Launch the Gradio interface if __name__ == "__main__": iface.launch()