biffboff commited on
Commit
18246c7
·
verified ·
1 Parent(s): abc35e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import json
@@ -9,7 +10,7 @@ logging.basicConfig(level=logging.INFO)
9
  logger = logging.getLogger("StrategyInterpreterSpace")
10
 
11
  # Load model and tokenizer
12
- model_name = "EleutherAI/gpt-j-6B"
13
  logger.info(f"Loading model '{model_name}'...")
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  model = AutoModelForCausalLM.from_pretrained(model_name)
@@ -150,7 +151,7 @@ JSON:"""
150
  logger.error(f"Error generating strategy: {e}")
151
  return f"Error generating strategy: {e}"
152
 
153
- iface = gr.Interface(
154
  fn=interpret_strategy,
155
  inputs=gr.inputs.Textbox(lines=10, placeholder="Enter your strategy description here..."),
156
  outputs="text",
@@ -169,7 +170,7 @@ iface_suggest = gr.Interface(
169
  description="Generate a unique trading strategy based on risk level and market type."
170
  )
171
 
172
- app = gr.TabbedInterface([iface, iface_suggest], ["Interpret Strategy", "Suggest Strategy"])
173
 
174
  if __name__ == "__main__":
175
  app.launch()
 
1
+ # app.py
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import json
 
10
  logger = logging.getLogger("StrategyInterpreterSpace")
11
 
12
  # Load model and tokenizer
13
+ model_name = "EleutherAI/gpt-neo-2.7B" # Updated model
14
  logger.info(f"Loading model '{model_name}'...")
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
151
  logger.error(f"Error generating strategy: {e}")
152
  return f"Error generating strategy: {e}"
153
 
154
+ iface_interpret = gr.Interface(
155
  fn=interpret_strategy,
156
  inputs=gr.inputs.Textbox(lines=10, placeholder="Enter your strategy description here..."),
157
  outputs="text",
 
170
  description="Generate a unique trading strategy based on risk level and market type."
171
  )
172
 
173
+ app = gr.TabbedInterface([iface_interpret, iface_suggest], ["Interpret Strategy", "Suggest Strategy"])
174
 
175
  if __name__ == "__main__":
176
  app.launch()