johann22 commited on
Commit
bf6712a
1 Parent(s): 889f93a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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: