| import gradio as gr | |
| from transformers import pipeline | |
| # Initialize the text-generation pipeline with the Georgian model | |
| pipe = pipeline("text-generation", model="ai-forever/mGPT-1.3B-georgian") | |
| # Define a function that will handle user input and generate a response | |
| def chatbot(input_text): | |
| response = pipe(input_text, max_length=50, num_return_sequences=1) | |
| return response[0]["generated_text"] | |
| # Create the Gradio interface | |
| demo = gr.Interface( | |
| fn=chatbot, | |
| inputs="text", | |
| outputs="text", | |
| title="Georgian Chatbot", | |
| description="This is a chatbot powered by mGPT-1.3B-georgian model." | |
| ) | |
| # Launch the app | |
| demo.launch() | |