Rooni commited on
Commit
e92393d
·
1 Parent(s): 156b9de

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import requests
3
+ import json
4
+ import os
5
+
6
+ def gpt3_turbo(image=None, text=None):
7
+ # Замените 'YOUR_API_KEY' на ваш API ключ от OpenAI
8
+ api_key = os.getenv("API_KEY")
9
+
10
+ # Формируем данные для отправки запроса
11
+ data = {}
12
+ if image:
13
+ data['image'] = image
14
+ if text:
15
+ data['text'] = text
16
+
17
+ # Отправляем запрос на API GPT-3.5-turbo
18
+ headers = {
19
+ 'Content-Type': 'application/json',
20
+ 'Authorization': f'Bearer {api_key}'
21
+ }
22
+ endpoint = os.getenv("BASE_URL")
23
+ response = requests.post(endpoint, headers=headers, json={"prompt": data, "max_tokens": 50})
24
+
25
+ if response.status_code == 200:
26
+ result = response.json()
27
+ return result['choices'][0]['text']
28
+ else:
29
+ return "Ошибка при получении ответа от модели."
30
+
31
+ # Создаем интерфейс Gradio
32
+ iface = gr.Interface(
33
+ fn=gpt3_turbo,
34
+ inputs=["text", "text"],
35
+ outputs="text",
36
+ title="GPT",
37
+ description="Помощь ученикам, решение заданий",
38
+ interpretation="default"
39
+ )
40
+
41
+ # Запускаем интерфейс
42
+ iface.launch()