Update app.py

app.py

@@ -23,13 +23,13 @@ dtype = torch.float16
 example_images = ["00000174_003.png", "00006596_000.png", "00006663_000.png", 
                   "00012976_002.png", "00018401_000.png", "00019799_000.png"]
 
-processor = AutoProcessor.from_pretrained("StanfordAIMI/CheXagent-8b", trust_remote_code=True)
-generation_config = GenerationConfig.from_pretrained("StanfordAIMI/CheXagent-8b")
+processor = AutoProcessor.from_pretrained("liuhaotian/llava-v1.6-mistral-7b", trust_remote_code=True)
+generation_config = GenerationConfig.from_pretrained("liuhaotian/llava-v1.6-mistral-7b")
 # model = AutoModelForCausalLM.from_pretrained("StanfordAIMI/CheXagent-8b", torch_dtype=dtype, trust_remote_code=True)
 
 @spaces.GPU
 def generate(image, prompt):
-    model = AutoModelForCausalLM.from_pretrained("StanfordAIMI/CheXagent-8b", torch_dtype=dtype, trust_remote_code=True).to(device)
+    model = AutoModelForCausalLM.from_pretrained("liuhaotian/llava-v1.6-mistral-7b", torch_dtype=dtype, trust_remote_code=True).to(device)
     if hasattr(image, "read"):
         image = Image.open(io.BytesIO(image.read())).convert("RGB")
     else: