Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,10 @@ import re
|
|
91 |
import torch
|
92 |
from transformers import pipeline
|
93 |
|
94 |
-
|
|
|
|
|
|
|
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.
|