use hugging face if there is ollama_model is not set

app.py

@@ -31,21 +31,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 def choose_model():
-    if os.getenv("HF_TOKEN"):
-        print("Using HuggingFace")
+    if os.getenv("OLLAMA_MODEL"):
+        return LiteLLMModel(
+            model_id=os.getenv("OLLAMA_MODEL"),
+            api_base=os.getenv("OLLAMA_ENDPOINT"),
+            api_key=os.getenv("OLLAMA_KEY"),
+        )
+    else:
         return HfApiModel(
             max_tokens=2096,
             temperature=0.5,
             model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
             custom_role_conversions=None,
         )
-    else:
-        print("Using Ollama")
-        return LiteLLMModel(
-            model_id=os.getenv("OLLAMA_MODEL"),
-            api_base=os.getenv("OLLAMA_ENDPOINT"),
-            api_key=os.getenv("OLLAMA_KEY"),
-        )
 
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)