Update API_KEY validator
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
from spaces import GPU
|
2 |
|
|
|
3 |
from threading import Thread
|
4 |
from transformers import Qwen2VLForConditionalGeneration, Qwen2VLProcessor, TextIteratorStreamer, AutoProcessor, BatchFeature
|
5 |
from qwen_vl_utils import process_vision_info
|
@@ -48,10 +49,13 @@ def infer(
|
|
48 |
def respond(
|
49 |
message,
|
50 |
history,
|
|
|
51 |
max_tokens,
|
52 |
temperature,
|
53 |
top_p,
|
54 |
):
|
|
|
|
|
55 |
if len(history) == 0 or history[0]['role'] != 'system':
|
56 |
messages = [{"role": "system", "content": """You are Softie, or 小软 in Chinese.
|
57 |
You are an intelligent assistant developed by the School of Software at Hefei University of Technology.
|
@@ -68,6 +72,7 @@ app = ChatInterface(
|
|
68 |
respond,
|
69 |
type="messages",
|
70 |
additional_inputs=[
|
|
|
71 |
Slider(minimum=1, maximum=2048, value=512, step=1, label="最大生成长度"),
|
72 |
Slider(minimum=0.01, maximum=4.0, value=0.75, step=0.01, label="温度系数(Temperature)"),
|
73 |
Slider(minimum=0.01, maximum=1.0, value=0.5, step=0.01, label="核取样系数(Top-p)"),
|
|
|
1 |
from spaces import GPU
|
2 |
|
3 |
+
from os import getenv
|
4 |
from threading import Thread
|
5 |
from transformers import Qwen2VLForConditionalGeneration, Qwen2VLProcessor, TextIteratorStreamer, AutoProcessor, BatchFeature
|
6 |
from qwen_vl_utils import process_vision_info
|
|
|
49 |
def respond(
|
50 |
message,
|
51 |
history,
|
52 |
+
api_key,
|
53 |
max_tokens,
|
54 |
temperature,
|
55 |
top_p,
|
56 |
):
|
57 |
+
if api_key != getenv("API_KEY"):
|
58 |
+
raise ValueError("Invalid API Key")
|
59 |
if len(history) == 0 or history[0]['role'] != 'system':
|
60 |
messages = [{"role": "system", "content": """You are Softie, or 小软 in Chinese.
|
61 |
You are an intelligent assistant developed by the School of Software at Hefei University of Technology.
|
|
|
72 |
respond,
|
73 |
type="messages",
|
74 |
additional_inputs=[
|
75 |
+
Textbox(label="API Key"),
|
76 |
Slider(minimum=1, maximum=2048, value=512, step=1, label="最大生成长度"),
|
77 |
Slider(minimum=0.01, maximum=4.0, value=0.75, step=0.01, label="温度系数(Temperature)"),
|
78 |
Slider(minimum=0.01, maximum=1.0, value=0.5, step=0.01, label="核取样系数(Top-p)"),
|