Changes

Dockerfile

@@ -4,13 +4,16 @@ FROM python:3.10-slim
 # Set the working directory inside the container
 WORKDIR /app
 
-# Set Hugging Face cache directories to a writable location
-ENV HF_HOME=/app/.cache
-ENV HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    && rm -rf /var/lib/apt/lists/*
 
 # Create cache directory and set permissions
-RUN mkdir -p /app/.cache/huggingface && \
-    chmod -R 777 /app/.cache
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
+# Set environment variable for Hugging Face cache
+ENV HF_HOME=/app/cache
 
 # Copy the requirements file to install dependencies
 COPY requirements.txt .
@@ -21,12 +24,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the application code
 COPY app.py .
 
-# Create a non-root user and switch to it
-RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
-USER appuser
-
-# Expose the port Flask will run on
-EXPOSE 8000
+# Expose port 7860 (default for Hugging Face Spaces)
+EXPOSE 7860
 
-# Command to run the Flask application
-CMD ["python", "app.py"]
+# Command to run the FastAPI app with uvicorn
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,15 +1,15 @@
-from flask import Flask, jsonify
+from fastapi import FastAPI
 from datasets import load_dataset
 import random
 import json
 
-app = Flask(__name__)
+app = FastAPI()
 
 # Load the dataset from Hugging Face
 dataset = load_dataset("UCSC-Admire/idiom-SFT-dataset-561-2024-12-06_00-40-30", split="train")
 
-@app.route('/api/idioms', methods=['GET'])
-def get_idioms():
+@app.get("/api/idioms")
+async def get_idioms():
     # Select 50 random idioms from the dataset
     idioms = random.sample(list(dataset), 50)
     # Extract required fields
@@ -26,7 +26,4 @@ def get_idioms():
             "example": item.get("sentence", ""),
             "definition": compound_type
         })
-    return jsonify(response)
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+    return response