Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import model_api import query_model
+import prompt_builder import build_prompt
 
 # -------------------------
 # Page Config