Updated

main.py

@@ -20,7 +20,7 @@ from transformers import pipeline
 # CONFIGURATION
 # ===============================================
 API_SECRET = "techdisciplesai404"
-MODEL_NAME = "google/flan-t5-large"
+MODEL_NAME = "meta-llama/Llama-2-13b-chat-hf"
 DEVICE = 0 if torch.cuda.is_available() else -1
 
 # ===============================================
@@ -42,7 +42,7 @@ try:
 
 
     hf_pipeline = pipeline(
-        "text2text-generation",
+        "text-generation-inference",
         model=MODEL_NAME,
         device=DEVICE,
         max_new_tokens=1024,