Initial commit

app.py

@@ -20,15 +20,15 @@ if not hf_token:
 login(token=hf_token)
 logger.debug("Hugging Face login successful")
 
-# لود مدل Qwen2-1.5B-Instruct
-logger.debug("Loading Qwen/Qwen2-1.5B-Instruct model")
+# لود مدل CodeLlama
+logger.debug("Loading meta-llama/CodeLlama-7b-Instruct-hf model")
 try:
-    tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B-Instruct", cache_dir="/tmp/hf_cache")
-    model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-1.5B-Instruct", cache_dir="/tmp/hf_cache", torch_dtype=torch.bfloat16, device_map="auto")
+    tokenizer = AutoTokenizer.from_pretrained("meta-llama/CodeLlama-7b-Instruct-hf", cache_dir="/tmp/hf_cache")
+    model = AutoModelForCausalLM.from_pretrained("meta-llama/CodeLlama-7b-Instruct-hf", cache_dir="/tmp/hf_cache", torch_dtype=torch.bfloat16, device_map="auto")
     code_gen = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512, temperature=0.6, top_p=0.95)
-    logger.debug("Qwen/Qwen2-1.5B-Instruct model loaded")
+    logger.debug("meta-llama/CodeLlama-7b-Instruct-hf model loaded")
 except Exception as e:
-    logger.error(f"Failed to load Qwen2-1.5B-Instruct model: {e}")
+    logger.error(f"Failed to load CodeLlama model: {e}")
     raise
 
 # آزادسازی حافظه
@@ -97,19 +97,6 @@ def get_history():
         logger.error(f"History retrieval error: {e}")
         return f"خطا در دریافت تاریخچه: {e}"
 
-# تابع دیباگ برای تست مدل
-def debug_model(text, language):
-    logger.debug(f"Debugging model with input: {text}, language: {language}")
-    try:
-        prompt = f"Write a complete, correct, and well-explained code in {language} to: {text}"
-        logger.debug(f"Generated prompt: {prompt}")
-        result = code_gen(prompt, max_new_tokens=512, temperature=0.6, top_p=0.95, do_sample=True)[0]['generated_text']
-        logger.debug(f"Generated output: {result}")
-        return f"Debug: Prompt: {prompt}\nOutput: {result}"
-    except Exception as e:
-        logger.error(f"Debug error: {e}")
-        return f"Debug error: {e}"
-
 # تولید کد
 def generate_code(text, language):
     logger.debug(f"Processing input: {text}, language: {language}")

requirements.txt

@@ -1,12 +1,11 @@
-datasets==2.14.5
-huggingface-hub>=0.19
-hf_xet>=1.0.0,<2.0.0
-hf-transfer>=0.1.4
-protobuf<4
-click<8.1
-pydantic==2.7.0
-torch==2.0.1
-numpy==1.24.3
-transformers==4.35.2
-gradio==4.31.0
-sentencepiece
+datasets
+huggingface_hub
+hf_xet
+hf-transfer
+protobuf
+click
+pydantic
+torch
+numpy
+transformers
+gradio