IbrahimHasani commited on
Commit
62b6525
1 Parent(s): 7228139

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
2
  from PIL import ImageFilter, Image
3
  from transformers import AutoModelForZeroShotImageClassification, AutoProcessor
4
  import torch
5
- import requests
6
 
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
@@ -68,11 +67,10 @@ def classify_image(image, candidate_labels):
68
 
69
  # Default values
70
  default_labels = "human with beverage,human,beverage"
71
- default_image_url = "F50tIRvbsAAKHzW.jpeg"
72
 
73
- # Download default image
74
- response = requests.get(default_image_url)
75
- default_image = Image.open(BytesIO(response.content))
76
 
77
  iface = gr.Interface(
78
  fn=classify_image,
 
2
  from PIL import ImageFilter, Image
3
  from transformers import AutoModelForZeroShotImageClassification, AutoProcessor
4
  import torch
 
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
 
67
 
68
  # Default values
69
  default_labels = "human with beverage,human,beverage"
70
+ default_image_path = "F50tIRvbsAAKHzW.jpeg"
71
 
72
+ # Load default image
73
+ default_image = Image.open(default_image_path)
 
74
 
75
  iface = gr.Interface(
76
  fn=classify_image,