Spaces:
Runtime error
Runtime error
File size: 1,265 Bytes
2fbf461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from flask import Flask, request, jsonify
from flask_cors import CORS
from transformers import pipeline
from transformers import AutoProcessor, AutoModelForCausalLM, AutoTokenizer
import requests
from PIL import Image
model_folder_path = r'C:\Users\ravit\OneDrive\Documents\nsfw_filter\nsfw_extension\models'
zsc_model_path = r'C:\Users\ravit\OneDrive\Documents\nsfw_filter\nsfw_extension\zscmodel'
candidate_labels = ["Violent", "Neutral", "Sexually explicit" ]
image_url = r"C:\Users\ravit\OneDrive\Documents\nsfw_filter\nsfw_extension\download.jpeg"
print(image_url)
tokenizer = AutoTokenizer.from_pretrained(model_folder_path)
model = AutoModelForCausalLM.from_pretrained(model_folder_path)
processor = AutoProcessor.from_pretrained(model_folder_path)
# image = Image.open(requests.get(image_url, stream=True).raw)
image = Image.open(image_url)
pixel_values = processor(images=image, return_tensors="pt").pixel_values
generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
classifier = pipeline("zero-shot-classification", model=zsc_model_path)
output = classifier(generated_caption, candidate_labels, multi_label=False)
print(output)
|