Update myapp.py

myapp.py

@@ -4,7 +4,7 @@ from transformers import AutoModel, AutoTokenizer
 from fastsafetensors import safe_load
 
 # Initialize the Flask app
-app = Flask(__name__)
+myapp = Flask(__name__)
 
 # Load the model and tokenizer using safe_load
 model_path = "https://huggingface.co/prompthero/openjourney-v4/blob/main/safety_checker/model.safetensors"  # Replace with your .safetensors file path
@@ -19,11 +19,11 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 # Load the model weights from safeload
 model = AutoModel.from_pretrained(model_name, state_dict=model_data)
 
-@app.route('/')
+@myapp.route('/')
 def index():
     return "Welcome to the AI Model API!"
 
-@app.route('/generate', methods=['POST'])
+@myapp.route('/generate', methods=['POST'])
 def generate_output():
     data = request.json
     prompt = data.get('prompt', 'Hello, world!')
@@ -39,4 +39,4 @@ def generate_output():
     return jsonify(outputs)
 
 if __name__ == "__main__":
-    app.run(host='0.0.0.0', port=5000)
+    myapp.run(host='0.0.0.0', port=5000)