Fix formating issue

src/app.py

@@ -27,7 +27,7 @@ def text_description_to_locations(description: str) -> list[str]:
 
     text = llm_request(messages)
 
-    text = text.split("assistant")[-1].strip()
+    text = text.strip()
 
     # return ["Paris, France", "Rome, Italy", "Barcelona, Spain", "Amsterdam, Netherlands"]
 
@@ -77,7 +77,7 @@ def describe_location_with_llm(location_name: str) -> str:
 
     text = llm_request(messages)
 
-    text = text.split("assistant")[-1].strip()
+    text = text.strip()
 
     return text
 
@@ -93,7 +93,7 @@ def llm_request(messages) -> str:
 
     text = pipe(messages, max_new_tokens=max_new_tokens)
 
-    return text
+    return text[-1]['content']
 
 async def main():
     st.write("""