fffiloni commited on
Commit
9e731de
1 Parent(s): c93d85e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -91,7 +91,10 @@ import re
91
  import torch
92
  from transformers import pipeline
93
 
94
- pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
 
 
 
95
 
96
  agent_maker_sys = f"""
97
  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.
 
91
  import torch
92
  from transformers import pipeline
93
 
94
+ zephyr_model = "HuggingFaceH4/zephyr-7b-beta"
95
+ mixtral_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
96
+
97
+ pipe = pipeline("text-generation", model=mixtral_model, torch_dtype=torch.bfloat16, device_map="auto")
98
 
99
  agent_maker_sys = f"""
100
  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.