Update app.py

app.py

@@ -7,15 +7,8 @@ import re
 
 # Model configurations
 MODELS = {
-    "Athena-R3X 8B": "Spestly/Athena-R3X-8B",
-    "Athena-R3X 4B": "Spestly/Athena-R3X-4B",
-    "Athena-R3 7B": "Spestly/Athena-R3-7B",
-    "Athena-3 3B": "Spestly/Athena-3-3B",
-    "Athena-3 7B": "Spestly/Athena-3-7B",
-    "Athena-3 14B": "Spestly/Athena-3-14B",
-    "Athena-2 1.5B": "Spestly/Athena-2-1.5B",
-    "Athena-1 3B": "Spestly/Athena-1-3B",
-    "Athena-1 7B": "Spestly/Athena-1-7B"
+    "20B": "openai/gpt-oss-20b",
+    "120B": "openai/gpt-oss-20b",
 }
 
 @spaces.GPU
@@ -38,10 +31,9 @@ def generate_response(model_id, conversation, user_message, max_length=512, temp
     # Build messages in proper chat format (OpenAI-style messages)
     messages = []
     system_prompt = (
-        "You are Athena, a helpful, harmless, and honest AI assistant. "
+        "You are GPT, a helpful, harmless, and honest AI assistant. "
         "You provide clear, accurate, and concise responses to user questions. "
         "You are knowledgeable across many domains and always aim to be respectful and helpful. "
-        "You are finetuned by Aayan Mishra"
     )
     messages.append({"role": "system", "content": system_prompt})
 
@@ -175,8 +167,8 @@ css = """
 }
 """
 
-with gr.Blocks(title="Athena Playground Chat", css=css, theme='NoCrypt/miku') as demo:
-    gr.Markdown("# 🚀 Athena Playground Chat")
+with gr.Blocks(title="GPT-OSS Playground Chat", css=css, theme='NoCrypt/miku') as demo:
+    gr.Markdown("# 🚀 GPT-OSS Playground Chat")
     gr.Markdown("*Powered by HuggingFace ZeroGPU*")
 
     # State to keep track of the conversation for the model
@@ -197,7 +189,7 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme='NoCrypt/miku') as
         model_choice = gr.Dropdown(
             label="📱 Model",
             choices=list(MODELS.keys()),
-            value="Athena-R3X 4B",
+            value="20B",
             info="Select which Athena model to use"
         )
         max_length = gr.Slider(
@@ -243,11 +235,6 @@ with gr.Blocks(title="Athena Playground Chat", css=css, theme='NoCrypt/miku') as
         inputs=[user_input]
     )
 
-    gr.Markdown("""
-    ### About the Thinking Tags
-    Some Athena models (particularly R3X series) include reasoning in `<think></think>` tags.
-    Click "Show reasoning" to see the model's thought process behind its answers.
-    """)
 
 if __name__ == "__main__":
     demo.launch(debug=True)  # Enable debug mode for better error reporting