Chris4K commited on
Commit
3a53d6f
·
verified ·
1 Parent(s): e6d8340

Update simple_sentiment.py

Browse files
Files changed (1) hide show
  1. simple_sentiment.py +1 -1
simple_sentiment.py CHANGED
@@ -71,7 +71,7 @@ class SimpleSentimentTool(Tool):
71
  raise RuntimeError(f"Failed to load any sentiment model: {str(e)}")
72
  return self._classifiers[model_id]
73
 
74
- def forward(self, text: str, model_key=None):
75
  """Process input text and return sentiment predictions."""
76
  try:
77
  # Determine which model to use
 
71
  raise RuntimeError(f"Failed to load any sentiment model: {str(e)}")
72
  return self._classifiers[model_id]
73
 
74
+ def forward(self, text: str, model_key="oliverguhr/german-sentiment-bert"):
75
  """Process input text and return sentiment predictions."""
76
  try:
77
  # Determine which model to use