EdBoy2202 commited on
Commit
feb93b6
·
verified ·
1 Parent(s): 394d5d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ from io import BytesIO
8
  import matplotlib.pyplot as plt
9
  import numpy as np
10
  from sklearn.preprocessing import LabelEncoder
 
 
11
  from huggingface_hub import hf_hub_download
12
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
13
  import torch
 
8
  import matplotlib.pyplot as plt
9
  import numpy as np
10
  from sklearn.preprocessing import LabelEncoder
11
+ from sklearn.feature_extraction.text import TfidfVectorizer
12
+ from sklearn.metrics.pairwise import cosine_similarity
13
  from huggingface_hub import hf_hub_download
14
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
15
  import torch