switch to base model

src/app.py

@@ -1,16 +1,19 @@
-from threading import Thread
 from typing import Optional
 import streamlit as st
 import pandas as pd
 import plotly.express as px
 import os
+import time
 import requests
 import torch
-from transformers import pipeline, TextStreamer, AsyncTextIteratorStreamer
+from transformers import pipeline
 
 @st.cache_resource
 def model_loader():
     model_name = "OrangeBottle/place_description_model_16_bit"
+    model_name = "unsloth/Llama-3.2-3B-Instruct" # Non Fine-Tuned Base Model
+
+    st.write(f"##### Using {model_name}")
 
     pipe = pipeline(task="text-generation", model=model_name, dtype=torch.bfloat16, device_map="cpu")
 
@@ -55,7 +58,6 @@ def search_location(location_name) -> Optional[dict[str, float]]:
     latitude = location['lat']
     longitude = location['lon']
 
-    import time
     time.sleep(1)  # To respect rate limiting
 
     return {