Upload 2 files

app.py

@@ -2,6 +2,11 @@ from fastapi import FastAPI
 
 app = FastAPI()
 
-@app.get("/")
-def read_root():
-    return {"message": "Fine-tuning script is running"}
+@app.post("/fine-tune")
+def fine_tune_model():
+    import subprocess
+    result = subprocess.run(["python", "fine_tune.py"], capture_output=True, text=True)
+    if result.returncode == 0:
+        return {"status": "Fine-tuning completed successfully!", "logs": result.stdout}
+    else:
+        raise HTTPException(status_code=500, detail=f"Error during fine-tuning: {result.stderr}")

dockerfile

@@ -19,9 +19,8 @@ RUN pip install --no-cache-dir --upgrade pip && \
 # Copy the application files
 COPY . .
 
-# Expose the port your app runs on (7860 is the default for Spaces)
+# Expose the port your app runs on
 EXPOSE 7860
 
-# Command to start the FastAPI app
+# Command to start the Python fine-tuning script
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-