feat: Add new model options for inference and update requirements for OAuth support in Gradio.

app.py

@@ -17,6 +17,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 MODELS = [
     "Local-Model-1",
     "Local-Model-2",
+    "Elfsong/VLM-iter_0000500",
+    "Elfsong/VLM-iter_0001000",
+    "Elfsong/VLM-iter_0001500",
 ]
 
 DATA_DIR = Path("logs")
@@ -56,11 +59,11 @@ def model_inference(user_message, history, model_name, system_message, max_token
     else:
         client = InferenceClient(token=token, model=model_name)
     
-    # 构建消息列表
+    # Construct message list
     history.append({"role": "user", "content": user_message})
     history.append({"role": "assistant", "content": ""})
     
-    # 构造 API 发送的消息（包含系统提示词）
+    # Construct API messages (including system prompt)
     api_messages = [{"role": "system", "content": system_message}] + history[:-1]
 
     try:
@@ -95,7 +98,7 @@ def model_inference(user_message, history, model_name, system_message, max_token
 with gr.Blocks() as demo:
     with gr.Sidebar():
         gr.Markdown("## Configuration")
-        # gr.LoginButton()
+        gr.LoginButton()
         
         # Local vLLM endpoint setting
         local_endpoint_a = gr.Textbox(

requirements.txt

@@ -1 +1,2 @@
 gradio
+gradio[oauth]