Update app.py

app.py

@@ -78,11 +78,11 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
             result.save(png_path)
             image = str(Path(png_path).resolve())
         return image
-    return None
+    return result
 
 def gen_fnseed(model_str, prompt, seed=1):
     if model_str == 'NA':
-        return None
+        return result
     try:
         loop = asyncio.new_event_loop()
         result = loop.run_until_complete(infer(model_str, prompt, seed, inference_timeout))
@@ -99,7 +99,7 @@ def gen_fnseed(model_str, prompt, seed=1):
 
 def gen_fnsix(model_str, prompt):
     if model_str == 'NA':
-        return None
+        return result
     noisesix = str(randint(1941, 2023)) #str(randint(0, 99999999999))
     return models_load[model_str](f'{prompt} {noisesix}')
 with gr.Blocks() as demo: