ddd

.dockerfile

@@ -0,0 +1,20 @@
+# Use the official Python image from the Docker Hub
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the requirements file into the container at /app
+COPY requirements.txt requirements.txt
+
+# Install the dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code into the container at /app
+COPY . .
+
+# Expose port 8000 for the FastAPI app
+EXPOSE 8000
+
+# Command to run the FastAPI app using Uvicorn
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]

.gitattributes

@@ -1 +1,3 @@
 *.pth filter=lfs diff=lfs merge=lfs -text
+*.pth"
+ filter=lfs diff=lfs merge=lfs -text

app.py

@@ -19,7 +19,7 @@ model.classifier[6] = nn.Linear(4096, 2)
 model.load_state_dict(torch.load('model_vgg16.pth' , map_location=device))
 
 model.eval()
-
+s
 transform = T.Compose([
     T.Resize((224, 224)),
     T.ToTensor(),