Update app.py

app.py

@@ -226,7 +226,7 @@ def handle_idle_state():
 
 # S.O.U.L. (Self-Organizing Universal Learning) Function
 class SOUL:
-    def __init__(self, model_name='EleutherAI/gpt-neox-20b'):
+    def __init__(self, model_name='EleutherAI/gpt-neox-2.7B'):
         self.tokenizer = GPTNeoXTokenizerFast.from_pretrained(model_name)
         self.model = GPTNeoXForCausalLM.from_pretrained(model_name)
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -266,6 +266,7 @@ def interact_with_soul(user_input):
     neox_response, emotional_response = soul.bridge_ai(user_input)
     return neox_response, emotional_response
 
+# Gradio interface setup
 iface = gr.Interface(
     fn=interact_with_soul,
     inputs="text",
@@ -274,4 +275,5 @@ iface = gr.Interface(
     description="Enter a prompt to interact with the S.O.U.L AI, which will generate a response and provide an emotional analysis."
 )
 
+# Launch the interface
 iface.launch()