TheFrenchDemos commited on
Commit
9bf9553
·
1 Parent(s): b556cc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -2,13 +2,13 @@
2
 
3
  from PIL import Image
4
  import numpy as np
5
- import io
6
- import faiss
7
 
8
- import requests
9
- import torch
10
 
11
- from request import get_ft, get_topk
12
  from flickrapi import FlickrAPI
13
 
14
  from flask import Flask, request, render_template, jsonify, send_from_directory
@@ -144,7 +144,7 @@ DEFAULT_PARAMS = {
144
 
145
  @app.route('/select_preset/<int:preset_id>')
146
  def select_preset(preset_id):
147
- if preset_id not in PRESET_IMAGES: #vérification de l'existence de preset_id dans le dictionnaire
148
  return jsonify({'error': 'Invalid preset ID'}), 400
149
 
150
  try:
 
2
 
3
  from PIL import Image
4
  import numpy as np
5
+ import io #manipule des images téléchargées via une URL ou une API
6
+ import faiss #librairie pour la recherche de similarité dans des vecteurs
7
 
8
+ import requests #télécharge des images et récupére une image depuis une URL
9
+ import torch #modèles de machine learning pour extraire les features d’images (vecteurs)
10
 
11
+ from request import get_ft, get_topk #get_ft extrait les vecteurs, get_topk trouve les k plus proches voisins
12
  from flickrapi import FlickrAPI
13
 
14
  from flask import Flask, request, render_template, jsonify, send_from_directory
 
144
 
145
  @app.route('/select_preset/<int:preset_id>')
146
  def select_preset(preset_id):
147
+ if preset_id not in PRESET_IMAGES: #vérifie l'existence de preset_id dans le dictionnaire
148
  return jsonify({'error': 'Invalid preset ID'}), 400
149
 
150
  try: