import gradio as gr
from transformers import pipeline

# part1 = hf_rSFLYgnNSgUWyBc
# part2 = BfOwDzWGSnfkeBVbYik

# Установка API ключа
# api_token = part1 + part2
# hf_api().token = api_token

# Инициализация пайплайна для генерации текста
pipe = pipeline("text-generation", model="meta-llama/CodeLlama-7b-hf")

def generate_text(prompt):
    # Генерация текста на основе введенного промпта
    generated_text = pipe(prompt, max_length=50, do_sample=True)[0]['generated_text']
    return generated_text

# Создание Gradio интерфейса
demo = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Text Generation with CodeLlama-7b-hf",
    description="Enter a prompt and see the generated text."
)

# Запуск Gradio интерфейса
demo.launch()