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