Update app.py

app.py

@@ -2,16 +2,14 @@ import gradio as gr
 import os 
 from pathlib import Path
 import argparse
-model_file = "Yi-6B.q4_k_m.gguf"
-if not os.path.isfile("Yi-6B.q4_k_m.gguf"):
-    os.system("wget -c https://huggingface.co/SamPurkis/Yi-6B-GGUF/resolve/main/Yi-6B.q4_k_m.gguf")
+model_file = "yi-6b.Q4_0.gguf"
+if not os.path.isfile(model_file):
+    os.system("wget -c https://huggingface.co/TheBloke/Yi-6B-GGUF/resolve/main/yi-6b.Q4_0.gguf")
 
 DEFAULT_MODEL_PATH = model_file
 
 from llama_cpp import Llama
-llm = Llama(model_path=model_file)
-old_tokenize = llm._model.tokenize
-llm._model.tokenize = lambda text, add_bos, spec: old_tokenize(text, False, spec)
+llm = Llama(model_path=model_file, model_type="mistral")
 
 
 def predict(input, chatbot, max_length, top_p, temperature, history):