Upload 2 files

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official Python runtime as the parent image
-FROM python:3.11-slim
+FROM python:3.10-slim
 
 # Set the working directory in the container
 WORKDIR /app
@@ -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"]

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")
@@ -79,5 +79,5 @@ def chatbot():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-if name == 'main':
-    app.run(host='0.0.0.0', port=8000)
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=8000)