Update app.py

app.py

@@ -1,13 +1,15 @@
-"""SusAI ©2025 Intern Labs. v1.0.5"""
+"""SusAI ©2025 Intern Labs. v1.0.1"""
 import os
 import gradio as gr
+
 from gradio_client import Client
 
 # Initialize Hugging Face Inference Client
-client = Client("HydroFlyer53/ThePickle", hf_token=os.environ["Key"])
+client = Client("HydroFlyer53/ThePickle", hf_token=os.environ.get("Key"))
 
 def chat_with_ai(message, history):
     """Function to get AI response from Hugging Face model."""
+
     result = client.predict(
         message=message,
         system_message=(
@@ -24,13 +26,8 @@ def chat_with_ai(message, history):
     )
     return result
 
-def startup_message():
-    return "SusAI ©2025 Intern Labs. v1.0.5"
-
+# Gradio Chat Interface
 demo = gr.ChatInterface(fn=chat_with_ai)
 
-demo.chatbot.((startup_message(), None))
-
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch()