Updated

main.py

@@ -21,7 +21,7 @@ from transformers import pipeline
 # CONFIGURATION
 # =====================================================
 API_SECRET = "techdisciplesai404"
-PRIMARY_MODEL = "openai-community/gpt2"
+PRIMARY_MODEL = "google/gemma-3-1b-it"
 FALLBACK_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 DEVICE = 0 if torch.cuda.is_available() else -1
 
@@ -56,7 +56,7 @@ def load_model(model_name, token=None):
             model=model_name,
             device=DEVICE,
             max_new_tokens=1024,
-            temperature=0.4,
+            temperature=0.3,
             top_p=0.9,
             repetition_penalty=1.15,
             do_sample=True,