Update app.py

app.py

@@ -22,6 +22,8 @@ agents =[
     "SOCIAL_MEDIA_MANAGER",
     "MEME_GENERATOR",
     "QUESTION_GENERATOR",
+    "IMAGE_GENERATOR",
+
 ]
 def generate(
         prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
@@ -44,7 +46,9 @@ def generate(
     if agent_name == "MEME_GENERATOR":
         agent = prompts.MEME_GENERATOR  
     if agent_name == "QUESTION_GENERATOR":
-        agent = prompts.QUESTION_GENERATOR      
+        agent = prompts.QUESTION_GENERATOR 
+    if agent_name == "IMAGE_GENERATOR":
+        agent = prompts.IMAGE_GENERATOR        
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2: