Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,9 @@ nltk.download('wordnet')
|
|
11 |
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
12 |
model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
13 |
|
|
|
|
|
|
|
14 |
def hashtag_generator(image):
|
15 |
raw_image = Image.fromarray(image).convert('RGB')
|
16 |
inputs = processor(raw_image, return_tensors="pt")
|
|
|
11 |
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
12 |
model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
|
13 |
|
14 |
+
stopwords = nltk.corpus.stopwords.words('english')
|
15 |
+
wn = nltk.WordNetLemmatizer()
|
16 |
+
|
17 |
def hashtag_generator(image):
|
18 |
raw_image = Image.fromarray(image).convert('RGB')
|
19 |
inputs = processor(raw_image, return_tensors="pt")
|