init

README.md

@@ -9,6 +9,7 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 short_description: Skywork/Skywork-o1-Open-PRM-Qwen-2.5-7B
+hf_oauth: true
 ---
 
 An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).

app.py

@@ -2,6 +2,8 @@ import spaces
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from __future__ import annotations
+from huggingface_hub import whoami
 
 # 定义系统提示语
 system_prompt = """你是 Skywork-o1，Skywork AI 开发的思维模型，擅长通过深度思考解决涉及数学、编码和逻辑推理的复杂问题。面对用户请求时，你首先会进行一段漫长而深入的思考过程，探索问题的可能解决方案。完成思考后，你会在回复中详细解释解决过程。"""
@@ -75,5 +77,23 @@ demo = gr.ChatInterface(
     # chatbot_style="default"
 )
 
+def hello(profile: gr.OAuthProfile | None) -> str:
+    if profile is None:
+        return "I don't know you."
+    return f"Hello {profile.name}"
+
+def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
+    if oauth_token is None:
+        return "Please deploy this on Spaces and log in to list organizations."
+    org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
+    return f"You belong to {', '.join(org_names)}."
+
+with gr.Blocks() as demo:
+    gr.LoginButton()
+    m1 = gr.Markdown()
+    m2 = gr.Markdown()
+    demo.load(hello, inputs=None, outputs=m1)
+    demo.load(list_organizations, inputs=None, outputs=m2)
+
 if __name__ == "__main__":
     demo.launch()