sunzijun
commited on
Commit
β’
a102655
1
Parent(s):
7b48c22
ADD: init repo
Browse files- app.py +53 -0
- requirements.txt +1 -0
app.py
ADDED
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env python
|
2 |
+
# -*- coding: utf-8 -*-
|
3 |
+
"""
|
4 |
+
@file : app.py
|
5 |
+
@author: sunzijun
|
6 |
+
@contact : sunzijun@pku.edu.cn
|
7 |
+
@date : 2022/12/7 1:08 δΈε
|
8 |
+
@version: 1.0
|
9 |
+
@desc :
|
10 |
+
"""
|
11 |
+
import os
|
12 |
+
|
13 |
+
from pyChatGPT import ChatGPT
|
14 |
+
|
15 |
+
|
16 |
+
def chat_hf(text, session_tokenz):
|
17 |
+
try:
|
18 |
+
# Save the session_token variable to the system environment
|
19 |
+
os.environ['SESSION_TOKEN'] = session_token
|
20 |
+
|
21 |
+
# Load the saved session_token value from the system environment
|
22 |
+
loaded_session_token = os.environ['SESSION_TOKEN']
|
23 |
+
|
24 |
+
api = ChatGPT(session_token)
|
25 |
+
resp = api.send_message(text)
|
26 |
+
|
27 |
+
api.refresh_auth() # refresh the authorization token
|
28 |
+
api.reset_conversation() # reset the conversation
|
29 |
+
xyz = resp['message']
|
30 |
+
except:
|
31 |
+
|
32 |
+
api = ChatGPT(session_tokenz)
|
33 |
+
resp = api.send_message(text)
|
34 |
+
|
35 |
+
api.refresh_auth() # refresh the authorization token
|
36 |
+
api.reset_conversation() # reset the conversation
|
37 |
+
xyz = resp['message']
|
38 |
+
|
39 |
+
return xyz
|
40 |
+
|
41 |
+
|
42 |
+
# @title GRadio for SDK api
|
43 |
+
|
44 |
+
import gradio as gr
|
45 |
+
|
46 |
+
gr.Interface(
|
47 |
+
chat_hf,
|
48 |
+
[gr.Textbox(label=' Input '),
|
49 |
+
gr.Textbox(label=' If it fails enter cusom session ')],
|
50 |
+
outputs=gr.outputs.Textbox(type="text", label="chatGPT response")
|
51 |
+
, title="" + ' ChatGpt π€π¬π» on hugginface. ' + "",
|
52 |
+
description="ChatGPT is a powerful dialog model trained by OpenAI").launch(
|
53 |
+
debug=True)
|
requirements.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
pyChatGPT
|