Upload app.py

app.py

@@ -161,7 +161,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # #model = AutoModelForCausalLM.from_pretrained(READER_MODEL_NAME)
 # model = AutoModelForCausalLM.from_pretrained(READER_MODEL_NAME).to(device)
 
-model_id = "neuralmagic/gemma-2-9b-it-quantized.w4a16"
+model_id = "mistralai/Mistral-7B-Instruct-v0.1"
 #model_id = "TheBloke/Gemma-2-7B-IT-GGUF"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(