Update app.py

app.py

@@ -11,9 +11,9 @@ from transformers import (
 )
 
 # Load model and processor
-model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-34b-hf")
+model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
 model.to("cuda:0")
-processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-34b-hf")
+processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
 
 @spaces.GPU(duration=240)
 def generate_caption(image):
@@ -36,7 +36,7 @@ def generate_caption(image):
 
 
 # Define the Gradio interface
-description = """This is a demo of [`llava-hf/llava-v1.6-34b-hf`](https://huggingface.co/llava-hf/llava-v1.6-34b-hf) hosted with ZeroGPU."""
+description = """This is a demo of [`llava-hf/llava-v1.6-mistral-7b-hf`](https://huggingface.co/llava-hf/llava-v1.6-mistral-7b-hf) hosted with ZeroGPU."""
 
 iface = gr.Interface(
     fn=generate_caption,