Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer
 from llava import LlavaForConditionalGeneration
 
 # Load the LLaVA model and tokenizer from the Hub
-model = LlavaForConditionalGeneration.from_pretrained("llava-hf/llava-1.5-7b-hf")
-tokenizer = AutoTokenizer.from_pretrained("llava-hf/llava-1.5-7b-hf")
+model = LlavaForConditionalGeneration.from_pretrained("liuhaotian/llava-v1.6-34b")
+tokenizer = AutoTokenizer.from_pretrained("liuhaotian/llava-v1.6-34b")
 
 # Define a function to generate a response given an input text and an optional image URL
 def generate_response(text, image_url=None):