Spaces:
Running
Running
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:
|