Update README.md

README.md

@@ -1,3 +1,14 @@
+---
+license: mit
+language:
+- en
+base_model:
+- openai/clip-vit-base-patch32
+pipeline_tag: image-classification
+tags:
+- aesthetics
+- creativity
+---
 
 # Aesthetic Scorer
 
@@ -24,8 +35,8 @@ import torch
 from PIL import Image
 
 # Load the model
-processor = CLIPProcessor.from_pretrained("YOUR_USERNAME/aesthetic-scorer")
-model = torch.load("YOUR_USERNAME/aesthetic-scorer/model.pt")
+processor = CLIPProcessor.from_pretrained("rsinema/aesthetic-scorer")
+model = torch.load("rsinema/aesthetic-scorer/model.pt")
 
 # Process an image
 image = Image.open("your_image.jpg")
@@ -38,5 +49,5 @@ with torch.no_grad():
 # Print results
 aesthetic_categories = ["Overall", "Quality", "Composition", "Lighting", "Color", "Depth of Field", "Content"]
 for category, score in zip(aesthetic_categories, scores):
-    print(f"{category}: {score.item():.2f}/10")
-```
+    print(f"{category}: {score.item():.2f}/5")
+```