Add File

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.9
+
+WORKDIR /code
+
+# Create cache directory and set permissions
+RUN mkdir -p /.cache/huggingface/hub && \
+    chmod -R 777 /.cache
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn","-b", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,75 @@
+import os
+from transformers import AutoModelForImageClassification, AutoFeatureExtractor
+import torch
+from flask_cors import CORS
+from flask import Flask, request, json
+from PIL import Image
+import requests
+from io import BytesIO
+from bs4 import BeautifulSoup
+from urllib.parse import urljoin
+
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+
+app = Flask(__name__)
+cors = CORS(app)
+
+# Define the model and feature extractor globally
+model = AutoModelForImageClassification.from_pretrained('carbon225/vit-base-patch16-224-hentai')
+feature_extractor = AutoFeatureExtractor.from_pretrained('carbon225/vit-base-patch16-224-hentai')
+
+@app.route("/", methods=["GET"])
+def default():
+    return json.dumps({"Server": "Working"})
+
+@app.route("/extractimages",methods=["GET"])
+def extract_images():
+    try:
+        src=request.args.get("src")
+        response = requests.get(src)
+        soup = BeautifulSoup(response.content,'html.parser')
+        img_urls=[]
+
+        img_tags = soup.select('div img')
+        for img_tag in img_tags:
+            img_url = urljoin(src, img_tag['src'])
+            img_urls.append(img_url)
+        return json.dumps({"images":img_urls})
+    except Exception as e:
+        return e
+
+@app.route("/predict", methods=["GET"])
+def predict():
+    try:
+        src = request.args.get("src")
+
+        # Download image from the provided URL
+        response = requests.get(src)
+        response.raise_for_status()  
+
+        # Open and preprocess the image
+        image = Image.open(BytesIO(response.content))
+        image = image.resize((128, 128))
+
+        # Extract features using the pre-trained feature extractor
+        encoding = feature_extractor(images=image.convert("RGB"), return_tensors="pt")
+
+        # Make a prediction using the pre-trained model
+        with torch.no_grad():
+            outputs = model(**encoding)
+            logits = outputs.logits
+
+        # Get the predicted class index and label
+        predicted_class_idx = logits.argmax(-1).item()
+        predicted_class_label = model.config.id2label[predicted_class_idx]
+
+        # Return the predictions
+        return json.dumps({"class": predicted_class_label})
+
+    except requests.exceptions.RequestException as e:
+        return json.dumps({"error": f"Request error: {str(e)}"})
+    except Exception as e:
+        return json.dumps({"error": f"An unexpected error occurred: {str(e)}"})
+
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,58 @@
+blinker==1.7.0
+bs4==0.0.1
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+filelock==3.13.1
+Flask==3.0.0
+Flask-Cors==4.0.0
+fsspec==2023.12.2
+huggingface-hub==0.20.1
+idna==3.6
+importlib-metadata==7.0.0
+itsdangerous==2.1.2
+Jinja2==3.1.2
+markdownify==0.11.6
+MarkupSafe==2.1.3
+mpmath==1.3.0
+networkx==3.2.1
+numpy==1.26.2
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.101
+nvidia-nvtx-cu12==12.1.105
+outcome==1.2.0
+packaging==23.2
+perplexity-api @ git+https://gitlab.com/robowaifudev/perplexity-api@0baebed39beeedc13e6281af5505f4c4221fdb5f
+Pillow==10.1.0
+PySocks==1.7.1
+PyYAML==6.0.1
+regex==2023.10.3
+requests==2.31.0
+safetensors==0.4.1
+selenium==4.12.0
+sortedcontainers==2.4.0
+sympy==1.12
+tokenizers==0.15.0
+torch==2.1.2
+tqdm==4.66.1
+transformers==4.36.2
+trio==0.22.2
+trio-websocket==0.10.4
+triton==2.1.0
+typing_extensions==4.9.0
+urllib3==2.1.0
+Werkzeug==3.0.1
+wsproto==1.2.0
+zipp==3.17.0
+gunicorn
+torch
+bs4