Update app.py

app.py

@@ -67,6 +67,31 @@ if user_input:
     text_to_speech(chatbot_response)
 
     text_to_speech(chatbot_response)
+    import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+
+# Use logging instead of print for more structured output
+def load_model():
+    try:
+        logging.info("Loading model...")
+        model_name = "declare-lab/tango-full"
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HUGGING_FACE_API_KEY)
+        logging.info("Model loaded successfully.")
+        return model, tokenizer
+    except Exception as e:
+        logging.error(f"Error loading model: {e}")
+        raise
+
+# Example usage in your Streamlit code
+if __name__ == "__main__":
+    try:
+        model, tokenizer = load_model()
+    except Exception as e:
+        logging.error(f"Application failed to start: {e}")
+