Update app.py

app.py

@@ -30,6 +30,10 @@ text_model.to(device).eval()
 autoencoder.to(device).eval()
 
 def generate_image(text_query):
-  return generate_image_from_text(text_query, text_tokenizer, text_model, autoencoder, biggan, device=device)
+  array = generate_image_from_text(text_query, text_tokenizer, text_model, autoencoder, biggan, device=device)
+  array = ((array + 1.0) / 2.0) * 256
+  array.clip(0, 255, out=array)
+  array = np.asarray(np.uint8(array), dtype=np.uint8)
+  return array
 
 gr.Interface(fn=generate_image, inputs="text", outputs="image").launch()