Mail.ruПочтаМой МирОдноклассникиВКонтактеИгрыЗнакомстваНовостиКалендарьОблакоЗаметкиВсе проекты

Помогите с ошибкой в коде

Игорь Перевощиков Ученик (34), открыт 3 недели назад
from transformers import GPT2LMHeadModel, GPT2Tokenizer, CLIPProcessor, CLIPVisionModelWithProjection
import torch
import requests
from io import BytesIO
from PIL import Image

# Загружаем модели и токенизаторы
gpt2_model = GPT2LMHeadModel.from_pretrained('gpt2')
gpt2_tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
clip_model = CLIPVisionModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")

# Токен вашего бота, полученный у @BotFather
TOKEN = 'YOUR_BOT_TOKEN_HERE'

# Команда /start
def start(update, context):
update.message.reply_text('Привет! Я твой новый бот. Что хочешь сделать?\n\n'
'/gen_image <описание> - создать изображение\n'
'/gen_text <начало текста> - продолжить текст\n'
'/gen_code <язык программирования> <задача> - сгенерировать код')

# Генерация изображения
def gen_image(update, context):
try:
description = update.message.text.split('/gen_image ', 1)[1]
image = generate_image(description)
bio = BytesIO()
image.save(bio, 'PNG')
bio.seek (0)
update.message.reply_photo(photo=bio)
except Exception as e:
update.message.reply_text(f'Что-то пошло не так: {e}')

# Генерация текста
def gen_text(update, context):
try:
prompt = update.message.text.split('/gen_text ', 1)[1]
text = generate_text(prompt)
update.message.reply_text(text)
except Exception as e:
update.message.reply_text(f'Что-то пошло не так: {e}')

# Генерация кода
def gen_code(update, context):
try:
args = update.message.text.split('/gen_code ', 1)[1]
language, task = args.split(' ', 1)
code = generate_code(language, task)
update.message.reply_text(code)
except Exception as e:
update.message.reply_text(f'Что-то пошло не так: {e}')

# Логика ошибок
def error(update, context):
print(f'Update "{update}" caused error "{context.error}"')

def main():
updater = Updater(TOKEN, use_context=True)
dp = updater.dispatcher

dp.add_handler(CommandHandler("start", start))
dp.add_handler(CommandHandler("gen_image", gen_image))
dp.add_handler(CommandHandler("gen_text", gen_text))
dp.add_handler(CommandHandler("gen_code", gen_code))

dp.add_error_handler(error)

updater.start_polling()
updater.idle()

if __name__ == '__main__':
main()

пишет
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
3 ответа
Maga Gur Ученик (142) 3 недели назад
Скорее тебе не сюда а к специалистам(
Гнездо Соловья Гуру (3597) 3 недели назад
Я ток через telebot бота делал(
Похожие вопросы