syntheticbot commited on
Commit
13df246
·
verified ·
1 Parent(s): 0a051b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -78,7 +78,7 @@ import requests
78
  # --- Configuration ---
79
  # This is the repository ID for the model on the Hugging Face Hub
80
  REPO_ID = "syntheticbot/clip-face-expression"
81
- FILENAME = "best_face_expression_model.safetensors"
82
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
83
 
84
  # The class names must be in alphabetical order as used during training
@@ -117,9 +117,9 @@ model.eval()
117
 
118
  # --- Run Prediction on an Example Image ---
119
  # Example image from the web
120
- url = "https://upload.wikimedia.org/wikipedia/commons/thumb/3/30/Expression_of_surprise.jpg/800px-Expression_of_surprise.jpg"
121
  try:
122
- image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
123
  except Exception as e:
124
  print(f"Could not load image from URL: {e}")
125
  exit()
 
78
  # --- Configuration ---
79
  # This is the repository ID for the model on the Hugging Face Hub
80
  REPO_ID = "syntheticbot/clip-face-expression"
81
+ FILENAME = "model.safetensors"
82
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
83
 
84
  # The class names must be in alphabetical order as used during training
 
117
 
118
  # --- Run Prediction on an Example Image ---
119
  # Example image from the web
120
+ url = "https://upload.wikimedia.org/wikipedia/commons/thumb/9/9d/Carol_Burnett_1958.JPG/250px-Carol_Burnett_1958.JPG"
121
  try:
122
+ image = Image.open(requests.get(url, stream=True,verify=False).raw).convert("RGB")
123
  except Exception as e:
124
  print(f"Could not load image from URL: {e}")
125
  exit()