Upload folder using huggingface_hub

app.py

@@ -1,3 +1,5 @@
+from IPython.display import clear_output, Image
+import argparse
 import numpy as np
 import random
 from PIL import Image
@@ -9,8 +11,8 @@ import torch
 import torchvision.transforms as transforms
 from PIL import Image
 
-device = torch.device('cpu')
-model = tag2text(pretrained='tag2text.pth', image_size=384, vit='swin_b')
+evice = torch.device('cpu')
+model = tag2text(pretrained='/content/recognize-anything/tag2text.pth', image_size=384, vit='swin_b')
 model.threshold = 0.68
 model.eval()
 
@@ -43,7 +45,7 @@ gr.Interface(
     inputs=inputs,
     outputs=outputs,
     title="Tags and Captioning using Tag2Text",
-    description="Upload an image and see the tags and captions in the output boxes",
+    description="Upload an image and see the results in text boxes.",
     live=True,  
     
 ).launch(share=True, enable_queue=True, debug=True)

requirements.txt

@@ -1,5 +1,5 @@
 timm==0.4.12
-transformers
+transformers==4.15.0
 fairscale==0.4.4
 pycocoevalcap
 torch