Update app.py

app.py

@@ -8,7 +8,7 @@ from io import BytesIO
 
 # Carregar o modelo Qwen-VL e o tokenizer
 tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-VL", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-VL", device_map="cuda", trust_remote_code=True).eval()
+model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-VL",load_in_4bit=True, device_map="auto", trust_remote_code=True).eval()
 
 def generate_predictions(image_input, text_input):
     # Inverter a imagem para corrigir o negativo