Update app.py

app.py

@@ -132,8 +132,7 @@ def main():
             + HABITAT_WORDS
         )
 
-        st.write(pipe(words))
-        image = pipe(words)["sample"][0]
+        image = pipe(words)["images"][0]
 
         plt.title(f"{last_generation + 1}th\n{words}.")
         plt.xticks([])