Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from PIL import Image
+import inference
 # Initialize session state to block re-running
 if 'has_run' not in st.session_state:
     st.session_state.has_run = False
@@ -21,5 +22,6 @@ if uploaded_image is not None:
         # Mark that the script has been run
         st.session_state.has_run = True
         st.write(task_prompt,"\n\n",text_input)
+        inference.demo()