Exemple #1
0
def post(message):
    if "files" in message.body.keys():
        labels = set()
        for file in message.body["files"]:
            label = animal.predict(file["url_private_download"])
            if label is not None:
                f = File()
                f.add(file["permalink"], label)
                labels.add(label)

        for l in labels:
            message.react(animal.get_emoji(l))
Exemple #2
0
import argparse
import os
import sys

sys.path.append(os.path.abspath("."))
import animal
from db import Database
from file import File

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-u", help="url")
    parser.add_argument("-a", help="animal", choices=animal.CLASS_NAMES)
    args = parser.parse_args()

    Database.initialise()
    file = File()
    file.add(url=args.u, animal=args.a)