Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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 |
-
|
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([
|
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()
|