.

app.py

@@ -1,4 +1,4 @@
-from my_app import create_app
+from create_app import create_app
 
 app = create_app()
 

create_app.py

@@ -5,7 +5,6 @@ from flask import render_template
 from transformers import LongformerTokenizer, AutoTokenizer, AutoModelForCausalLM, LongformerConfig
 from models.model import *
 from utils.util_func import *
-from safetensors.torch import load_file
 
 db = SQLAlchemy()
 login_manager = LoginManager()

inference/infer_single.py

@@ -1,7 +1,7 @@
 from utils.data_utils import *
 from utils.prompts import *
 import torch
-from my_app import * 
+from ..create_app import * 
 
 def replace_single_newlines(text):
     return re.sub(r'(?<!\n)\n(?!\n)', '\\\\n\\\\n', text)