walaa2022 commited on
Commit
da9c941
·
verified ·
1 Parent(s): 6d12d9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,7 +47,7 @@ class FinancialAnalyzer:
47
  self.financial_analyzer = pipeline(
48
  "text-classification",
49
  model="ProsusAI/finbert",
50
- return_all_scores=True
51
  )
52
  logger.info("FinBERT initialized successfully")
53
 
@@ -90,7 +90,7 @@ class FinancialAnalyzer:
90
 
91
  response = self.strategic_analyzer(
92
  prompt,
93
- max_length=1000,
94
  temperature=0.7
95
  )
96
  return response[0]['generated_text']
@@ -121,7 +121,7 @@ class FinancialAnalyzer:
121
 
122
  response = self.recommendation_generator(
123
  prompt,
124
- max_length=800,
125
  temperature=0.6
126
  )
127
  return response[0]['generated_text']
 
47
  self.financial_analyzer = pipeline(
48
  "text-classification",
49
  model="ProsusAI/finbert",
50
+ top_k= None
51
  )
52
  logger.info("FinBERT initialized successfully")
53
 
 
90
 
91
  response = self.strategic_analyzer(
92
  prompt,
93
+ max_length=1500,
94
  temperature=0.7
95
  )
96
  return response[0]['generated_text']
 
121
 
122
  response = self.recommendation_generator(
123
  prompt,
124
+ max_length=1500,
125
  temperature=0.6
126
  )
127
  return response[0]['generated_text']