Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ agents =[
|
|
23 |
"MEME_GENERATOR",
|
24 |
"QUESTION_GENERATOR",
|
25 |
"IMAGE_GENERATOR",
|
|
|
26 |
|
27 |
]
|
28 |
def generate(
|
@@ -48,7 +49,9 @@ def generate(
|
|
48 |
if agent_name == "QUESTION_GENERATOR":
|
49 |
agent = prompts.QUESTION_GENERATOR
|
50 |
if agent_name == "IMAGE_GENERATOR":
|
51 |
-
agent = prompts.IMAGE_GENERATOR
|
|
|
|
|
52 |
system_prompt=agent
|
53 |
temperature = float(temperature)
|
54 |
if temperature < 1e-2:
|
|
|
23 |
"MEME_GENERATOR",
|
24 |
"QUESTION_GENERATOR",
|
25 |
"IMAGE_GENERATOR",
|
26 |
+
"HUGGINGFACE_FILE_DEV",
|
27 |
|
28 |
]
|
29 |
def generate(
|
|
|
49 |
if agent_name == "QUESTION_GENERATOR":
|
50 |
agent = prompts.QUESTION_GENERATOR
|
51 |
if agent_name == "IMAGE_GENERATOR":
|
52 |
+
agent = prompts.IMAGE_GENERATOR
|
53 |
+
if agent_name == "HUGGINGFACE_FILE_DEV":
|
54 |
+
agent = prompts.HUGGINGFACE_FILE_DEV
|
55 |
system_prompt=agent
|
56 |
temperature = float(temperature)
|
57 |
if temperature < 1e-2:
|