PrachiY commited on
Commit
7e3bdf6
·
verified ·
1 Parent(s): 7b9b0a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ transform_input_image = transforms.Compose([
29
 
30
  def load_model_from_huggingface():
31
  model_filename = "ckpt_clothing_resnet50.pth"
32
- model_path = hf_hub_download(repo_id="austin-carnahan/c2d-clothing1M", filename=model_filename)
33
 
34
  # Load checkpoint and check keys
35
  checkpoint = torch.load(model_path, map_location=torch.device("cpu"), weights_only=False)
@@ -78,7 +78,7 @@ interface = gr.Interface(
78
  fn=gradio_inference,
79
  inputs=gr.Image(type="pil"),
80
  outputs="text",
81
- title="Clothing Image Classifier",
82
  description="Upload an image of clothing and get a predicted class with confidence score.",
83
  examples=["images/tshirt.jpg", "images/pants.jpg"]
84
  )
 
29
 
30
  def load_model_from_huggingface():
31
  model_filename = "ckpt_clothing_resnet50.pth"
32
+ model_path = hf_hub_download(repo_id="PrachiY/image-classification-model-space", filename=model_filename)
33
 
34
  # Load checkpoint and check keys
35
  checkpoint = torch.load(model_path, map_location=torch.device("cpu"), weights_only=False)
 
78
  fn=gradio_inference,
79
  inputs=gr.Image(type="pil"),
80
  outputs="text",
81
+ title="image-classification-model",
82
  description="Upload an image of clothing and get a predicted class with confidence score.",
83
  examples=["images/tshirt.jpg", "images/pants.jpg"]
84
  )