Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import yaml
+import litellm
 import logging
 from dotenv import load_dotenv
 from huggingface_hub import login