update

app.py

@@ -1,3 +1,5 @@
+import os
+import sys
 import argparse
 import gradio as gr
 
@@ -90,15 +92,19 @@ def main(args):
 
 
 if __name__ == "__main__":
+    intent_predict_model_weights_path = os.path.join(sys.path[0], "model_weights/Tiger_t5_base_encoder.pth")
+    text_dialog_model_weights_path = os.path.join(sys.path[0], "model_weights/Tiger_DialoGPT_medium.pth")
+    text2image_model_weights_path = os.path.join(sys.path[0], "model_weights/stable-diffusion-2-1-realistic")
+    
     parser = argparse.ArgumentParser()
     
     parser.add_argument('--intent_predict_model_name', type=str, default="t5-base")
-    parser.add_argument('--intent_predict_model_weights_path', type=str, default="model_weights/Tiger_t5_base_encoder.pth")
+    parser.add_argument('--intent_predict_model_weights_path', type=str, default=intent_predict_model_weights_path)
 
     parser.add_argument('--text_dialog_model_name', type=str, default="microsoft/DialoGPT-medium")
-    parser.add_argument('--text_dialog_model_weights_path', type=str, default="model_weights/Tiger_DialoGPT_medium.pth")
+    parser.add_argument('--text_dialog_model_weights_path', type=str, default=text_dialog_model_weights_path)
     
-    parser.add_argument('--text2image_model_weights_path', type=str, default="model_weights/stable-diffusion-2-1-realistic")
+    parser.add_argument('--text2image_model_weights_path', type=str, default=text2image_model_weights_path)
 
     parser.add_argument('--device', default="cuda:6")