Initial commit

app.py

@@ -32,12 +32,12 @@ except Exception as e:
     raise
 
 # لود مدل ترجمه
-logger.debug("Loading facebook/m2m100_418M model")
+logger.debug("Loading Helsinki-NLP/opus-mt-fa-en model")
 try:
-    translate_tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M", cache_dir="/tmp/hf_cache")
-    translate_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M", cache_dir="/tmp/hf_cache")
-    translate_gen = pipeline("translation", model=translate_model, tokenizer=translate_tokenizer, src_lang="fa", tgt_lang="en", max_length=300)
-    logger.debug("facebook/m2m100_418M model loaded")
+    translate_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-fa-en", cache_dir="/tmp/hf_cache")
+    translate_model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-fa-en", cache_dir="/tmp/hf_cache")
+    translate_gen = pipeline("translation", model=translate_model, tokenizer=translate_tokenizer, max_length=300)
+    logger.debug("Helsinki-NLP/opus-mt-fa-en model loaded")
 except Exception as e:
     logger.error(f"Failed to load translation model: {e}")
     raise
@@ -140,8 +140,11 @@ def translate_and_prompt(text, language):
 def generate_code(prompt):
     logger.debug(f"Generating code with prompt: {prompt}")
     try:
-        result = code_gen(prompt, max_length=512, num_beams=5)
+        result = code_gen(prompt, max_length=512, num_beams=5, do_sample=False)
         code_output = result[0]['generated_text']
+        if not code_output.strip():
+            logger.warning("Generated code is empty")
+            return "خطا: کد تولیدشده خالی است"
         logger.debug(f"Generated code: {code_output}")
         torch.cuda.empty_cache()
         return code_output