File size: 1,375 Bytes
d3bb606
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
27
28
29
30
31
32
33
34
35
36
37
38
39
import openai
from transformers import AutoModelForCausalLM, AutoTokenizer

# Установите API-ключ от OpenAI
API_KEY = api_key

def translate_text(text, source_lang, target_lang):
    # Задайте модель и токенизатор
    model_name = "gpt-3.5-turbo"
    tokenizer = AutoTokenizer.from_pretrained(model_name)
    model = AutoModelForCausalLM.from_pretrained(model_name)

    # Приведите текст к формату ввода для GPT-3
    input_text = f"Translate the following text from {source_lang} to {target_lang}: {text}"

    # Сделайте запрос к OpenAI GPT-3
    response = openai.Completion.create(
        engine="text-davinci-003",
        prompt=tokenizer.encode(input_text, return_tensors="pt"),
        max_tokens=100,
        temperature=0.7,
        n=1,
        stop=None,
    )

    # Получите перевод и раскодируйте его
    translated_text = tokenizer.decode(response.choices[0].text.strip())

    return translated_text

# Установите API-ключ
openai.api_key = API_KEY

# Пример использования
text = "Привет, как дела?"
translated_text = translate_text(text, source_lang="ru", target_lang="en")
print(f"Оригинальный текст: {text}")
print(f"Переведенный текст: {translated_text}")