diff --git a/py/api.py b/py/api.py
index 8ed3afb..413f951 100644
--- a/py/api.py
+++ b/py/api.py
@@ -2,6 +2,7 @@ import requests
 import json
 from transformers import AutoTokenizer, LlamaForCausalLM
 
+
 class API:
     @staticmethod
     def process_text_transformers(prompt, model):
@@ -46,4 +47,4 @@ class API:
             answer = self.process_text_transformers(message, "meta-llama/Meta-Llama-3.1-8B")
         else:
             return "Invalid choice"
-        return answer
\ No newline at end of file
+        return answer
diff --git a/py/static/index.html b/py/static/index.html
index 087940a..d688702 100644
--- a/py/static/index.html
+++ b/py/static/index.html
@@ -4,7 +4,7 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <link rel="stylesheet" href="styles.css">
+    <link rel="stylesheet" href="/static/styles.css">
     <title>AI Assistant</title>
 </head>
 
@@ -81,10 +81,10 @@
         <!-- Input section: Where user input is provided -->
         <div class="input">
             <input type="text" placeholder="Type your message here..." />
-            <button><img src="img/microphone.svg" alt="microphone"></button>
+            <button><img src="/static/img/microphone.svg" alt="microphone"></button>
             <button>Send</button>
         </div>
     </div>
 </body>
 
-</html>
\ No newline at end of file
+</html>
diff --git a/py/web_flask.py b/py/web_flask.py
index 3cc2681..2ac2ce9 100644
--- a/py/web_flask.py
+++ b/py/web_flask.py
@@ -1,4 +1,4 @@
-from flask import Flask
+from flask import Flask, send_from_directory
 
 app = Flask(__name__)
 
@@ -7,5 +7,6 @@ app = Flask(__name__)
 def index():
     return app.send_static_file('index.html')
 
+
 if __name__ == '__main__':
-    app.run()
+    app.run(debug=True)
