Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import json
@@ -9,7 +10,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("StrategyInterpreterSpace")
 
 # Load model and tokenizer
-model_name = "EleutherAI/gpt-j-6B"
+model_name = "EleutherAI/gpt-neo-2.7B"  # Updated model
 logger.info(f"Loading model '{model_name}'...")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
@@ -150,7 +151,7 @@ JSON:"""
         logger.error(f"Error generating strategy: {e}")
         return f"Error generating strategy: {e}"
 
-iface = gr.Interface(
+iface_interpret = gr.Interface(
     fn=interpret_strategy,
     inputs=gr.inputs.Textbox(lines=10, placeholder="Enter your strategy description here..."),
     outputs="text",
@@ -169,7 +170,7 @@ iface_suggest = gr.Interface(
     description="Generate a unique trading strategy based on risk level and market type."
 )
 
-app = gr.TabbedInterface([iface, iface_suggest], ["Interpret Strategy", "Suggest Strategy"])
+app = gr.TabbedInterface([iface_interpret, iface_suggest], ["Interpret Strategy", "Suggest Strategy"])
 
 if __name__ == "__main__":
     app.launch()