File size: 491 Bytes
0307592 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import pipeline
def get_translation_pipeline():
# Use a public model that doesn't require login
model_name = "Salesforce/codet5-base-multi-sum"
return pipeline("text2text-generation", model=model_name, tokenizer=model_name)
def translate_cobol_to_csharp(pipe, cobol_code: str) -> str:
prompt = f"Translate this COBOL code to C#:\n\n{cobol_code}"
result = pipe(prompt, max_length=512, do_sample=False)[0]["generated_text"]
return result
|