Update app.py

app.py

@@ -18,8 +18,9 @@ OLLAMA_SERVICE_THREAD.start()
 print("Giving ollama serve a moment")
 time.sleep(10)
 
-# Modify the model to what you want
-model = "gemma2"
+# Uncomment and modify the model to what you want locally
+# model = "moondream" 
+model = os.environ.get("MODEL")
 
 subprocess.run(f"~/ollama pull {model}", shell=True)
 
@@ -30,13 +31,10 @@ from ollama import Client
 client = Client(host='http://localhost:11434', timeout=120)
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
-MODEL_ID = os.environ.get("MODEL_ID", "google/gemma-2-9b-it")
-MODEL_NAME = MODEL_ID.split("/")[-1]
 
 TITLE = "<h1><center>ollama-Chat</center></h1>"
 
 DESCRIPTION = f"""
-<h3>MODEL: <a href="https://hf.co/{MODEL_ID}">{MODEL_NAME}</a></h3>
 <center>
 <p>Feel free to test models with ollama.
 <br>