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
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"""
|