Spaces:

Kluuking
/

ImageClassification_CatVsDogs

Runtime error

Kluuking commited on Feb 20, 2023

Commit

619dffd

1 Parent(s): a7764c4

added pip to install torch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import requests
 import gradio as gr
-import pytorch as torch
 from timm import create_model
 from timm.data import resolve_data_config
 from timm.data.transforms_factory import create_transform

 import requests
+pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
 import gradio as gr
+import torch
 from timm import create_model
 from timm.data import resolve_data_config
 from timm.data.transforms_factory import create_transform