Initial commit

app.py

@@ -20,15 +20,15 @@ if not hf_token:
 login(token=hf_token)
 logger.debug("Hugging Face login successful")
 
-# لود مدل Gemma 3 1B
-logger.debug("Loading google/gemma-3-1b-it model")
+# لود مدل Qwen2-1.5B-Instruct
+logger.debug("Loading Qwen/Qwen2-1.5B-Instruct model")
 try:
-    tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-1b-it", cache_dir="/tmp/hf_cache")
-    model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it", cache_dir="/tmp/hf_cache", torch_dtype=torch.bfloat16, device_map="auto")
+    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")
     code_gen = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512, temperature=0.6, top_p=0.95)
-    logger.debug("google/gemma-3-1b-it model loaded")
+    logger.debug("Qwen/Qwen2-1.5B-Instruct model loaded")
 except Exception as e:
-    logger.error(f"Failed to load Gemma 3 1B model: {e}")
+    logger.error(f"Failed to load Qwen2-1.5B-Instruct model: {e}")
     raise
 
 # آزادسازی حافظه