Update app.py

app.py

@@ -1,7 +1,7 @@
 # Dependencies
 import streamlit as st
 from huggingface_hub import InferenceClient
-import yaml
+import os
 
 def create_image(prompt):
     """