Update app.py

app.py

@@ -14,7 +14,8 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 image_size = 384
 
 normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
-                                std=[0.229, 0.224, 0.225])
+                                std=[0.229, 0.224, 0.225
+])
 transform = transforms.Compose([transforms.Resize((image_size, image_size)),transforms.ToTensor(),normalize])
 
 #######Tag2Text Model
@@ -41,7 +42,8 @@ def inference(raw_image, model_n , input_tag):
     image = transform(raw_image).unsqueeze(0).to(device) 
     if model_n == 'Recognize Anything Model':  
         model = model_ram
-        tags, tags_chinese = model.generate_tag(image)
+        with torch.no_grad():
+            tags, tags_chinese = model.generate_tag(image)
         return tags[0],tags_chinese[0], 'none'
     else:
         model = model_tag2text