Upload 5 files

Dockerfile

@@ -25,9 +25,13 @@ COPY static/script.js static/
 # Make port 8000 available to the world outside this container
 EXPOSE 8000
 
-# Define environment variable
+# Define environment variables
 ENV FLASK_APP=app.py
 ENV FLASK_RUN_HOST=0.0.0.0
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
+
+# Create the cache directory
+RUN mkdir -p /app/.cache/huggingface/hub
 
 # Run app.py when the container launches
-CMD ["flask", "run", "--port=8000"]
+CMD ["flask", "run", "--port=8000"]

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 import mysql.connector
 from groq import Groq
 
-app = Flask(name)
+app = Flask(__name__)
 
 # Initialize the text generation pipeline
 pipe = pipeline("text-generation", model="defog/sqlcoder-7b-2")
@@ -80,4 +80,4 @@ def chatbot():
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+    app.run(host='0.0.0.0', port=8000)