tommy24 commited on
Commit
b74f28e
·
1 Parent(s): d51fa15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -2
app.py CHANGED
@@ -1,12 +1,35 @@
1
  import gradio as gr
 
2
  import os
3
 
4
  def function(Textbox,Textbox2):
5
  target = os.environ.get("target")
 
 
 
 
6
  if Textbox2 == target:
7
- return "Yes"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  else:
9
- return "No"
10
 
11
  inputs = [
12
  gr.inputs.Textbox(label="Textbox",type="text"),
 
1
  import gradio as gr
2
+ import requests
3
  import os
4
 
5
  def function(Textbox,Textbox2):
6
  target = os.environ.get("target")
7
+ target2 = os.environ.get("target2")
8
+ model = os.environ.get("model")
9
+ hrc = os.environ.get("hrc")
10
+ content = os.environ.get("content")
11
  if Textbox2 == target:
12
+ payload = {
13
+ "model": "gpt-3.5-turbo",
14
+ "messages": [{"role": "system", "content": content},{"role": "user", "content": user}],
15
+ "temperature" : 1.0,
16
+ "top_p":1.0,
17
+ "n" : 1,
18
+ "stream": False,
19
+ "presence_penalty":0,
20
+ "frequency_penalty":0,
21
+ }
22
+
23
+ headers = {
24
+ "Content-Type": "application/json",
25
+ "Authorization": f"Bearer {target2}"
26
+ }
27
+
28
+ response = requests.post(hrc, headers=headers, json=payload, stream=False)
29
+ response = response.json()
30
+ return response["choices"][0]["message"]["content"]
31
  else:
32
+ return "Failed"
33
 
34
  inputs = [
35
  gr.inputs.Textbox(label="Textbox",type="text"),