Update app.py

app.py

@@ -1,4 +1,4 @@
-# import gradio as gr
+import gradio as gr
 import random
 import os
 from typing import Tuple , Dict
@@ -114,7 +114,7 @@ def predict(img= None , link:str = None) -> str:
             img = Image.fromarray(img)
             img = transform(img)
             img = img.unsqueeze(0)
-            answer += str((torch.argmax(model(img.to(device)))).cpu().numpy())
+            answer += str((torch.argmax(model(img.to(device)))).numpy())
 
         return answer , imgss