Hugging Face's logo

Spaces:
Alexvatti
/
Automatic-File-Renaming
Sleeping

App Files Community
Alexvatti commited on
Commit
4d3e1ee
·
verified ·
1 Parent(s): e332e0e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -40,7 +40,7 @@ llm = ChatOpenAI(
40
  )
41
 
42
  # Load the CLIP model for image feature extraction
43
- clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
44
  clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
45
 
46
  # Function to generate a more appropriate name based on content
 
40
  )
41
 
42
  # Load the CLIP model for image feature extraction
43
+ clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32",from_tf=True)
44
  clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
45
 
46
  # Function to generate a more appropriate name based on content