t

app.py

@@ -74,9 +74,11 @@ headers = {"Authorization": f"Bearer {os.environ['API_TOKEN']}"}
 
     
 
-def f(context, temperature, top_p, max_length, model_idx):
+def f(context, temperature, top_p, max_length, model_idx, SPACE_VERIFICATION_KEY):
     try:
 
+        if os.environ['SPACE_VERIFICATION_KEY'] != SPACE_VERIFICATION_KEY:
+            return "invalid SPACE_VERIFICATION_KEY; see project secrets to view key"
 
         # maybe try "0" instead or 1, or "1"
         # use GPT-J-6B
@@ -148,6 +150,8 @@ iface = gr.Interface(f, [
     gr.inputs.Slider(
         minimum=20, maximum=512, default=30, label="max length"),
     gr.inputs.Dropdown(["GPT-J-6B", "GPT-2"], type="index", label="model"),
+    gr.inputs.Textbox(lines=1, placeholder="space key", default="xxxxxxxx", label="space verification key")
+
 ], outputs="text", title=title, examples=examples)
 iface.launch()  # enable_queue=True