Rooni commited on
Commit
d3bb606
·
1 Parent(s): d5275c2

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
+
4
+ # Установите API-ключ от OpenAI
5
+ API_KEY = api_key
6
+
7
+ def translate_text(text, source_lang, target_lang):
8
+ # Задайте модель и токенизатор
9
+ model_name = "gpt-3.5-turbo"
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
11
+ model = AutoModelForCausalLM.from_pretrained(model_name)
12
+
13
+ # Приведите текст к формату ввода для GPT-3
14
+ input_text = f"Translate the following text from {source_lang} to {target_lang}: {text}"
15
+
16
+ # Сделайте запрос к OpenAI GPT-3
17
+ response = openai.Completion.create(
18
+ engine="text-davinci-003",
19
+ prompt=tokenizer.encode(input_text, return_tensors="pt"),
20
+ max_tokens=100,
21
+ temperature=0.7,
22
+ n=1,
23
+ stop=None,
24
+ )
25
+
26
+ # Получите перевод и раскодируйте его
27
+ translated_text = tokenizer.decode(response.choices[0].text.strip())
28
+
29
+ return translated_text
30
+
31
+ # Установите API-ключ
32
+ openai.api_key = API_KEY
33
+
34
+ # Пример использования
35
+ text = "Привет, как дела?"
36
+ translated_text = translate_text(text, source_lang="ru", target_lang="en")
37
+ print(f"Оригинальный текст: {text}")
38
+ print(f"Переведенный текст: {translated_text}")