Update app.py

app.py

@@ -91,7 +91,10 @@ import re
 import torch
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
+zephyr_model = "HuggingFaceH4/zephyr-7b-beta"
+mixtral_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
+
+pipe = pipeline("text-generation", model=mixtral_model, torch_dtype=torch.bfloat16, device_map="auto")
 
 agent_maker_sys = f"""
 You are an AI whose job is to help users create their own music which its genre will reflect the character or scene from an image described by users.