S01Nour commited on
Commit
c924863
·
1 Parent(s): b74a137

feat: Implement generation model manager for Hugging Face Seq2Seq models and add a debug script for testing generation.

Browse files
Files changed (1) hide show
  1. services/generate_model_manager.py +1 -1
services/generate_model_manager.py CHANGED
@@ -59,7 +59,7 @@ class GenerateModelManager:
59
  def _format_input(self, topic: str, position: str) -> str:
60
  """Format input for the model"""
61
  # Standard format for argument generation
62
- return f"topic: {topic} stance: {position}"
63
 
64
  def generate(self, topic: str, position: str, max_length: int = 128, num_beams: int = 4) -> str:
65
  """Generate argument for a topic and position"""
 
59
  def _format_input(self, topic: str, position: str) -> str:
60
  """Format input for the model"""
61
  # Standard format for argument generation
62
+ return f"generate argument: topic: {topic} stance: {position}"
63
 
64
  def generate(self, topic: str, position: str, max_length: int = 128, num_beams: int = 4) -> str:
65
  """Generate argument for a topic and position"""