czkaiweb commited on
Commit
8bd38e1
·
1 Parent(s): a13a427

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -44,9 +44,6 @@ def inference(input_image):
44
  probabilities = torch.nn.functional.softmax(output[0], dim=0)
45
 
46
  # Read the categories
47
- with open("artist_classes.txt", "r") as f:
48
- categories = [s.strip() for s in f.readlines()]
49
-
50
  categories = {
51
  0:"vanGogh",
52
  1:"Monet",
@@ -69,7 +66,7 @@ outputs = gr.Label(type="confidences",num_top_classes=5,label='Prediction')
69
  title = "Artist Classifier"
70
  description = "The classifier is a demo classifier to predict the painter using fine-tuned EfficientNet. It allows you to identify the creator of a painting among Van Gogh, Claude Monet, Leonardo da Vinci, Rembrandt, Pablo Picasso, and Salvador Dali. Just upload the image to the left blank box and click the Submit button. A list of confidence will be displayed. Following the link below to find a related work that helps to create your own paintings following the style of painters"
71
 
72
- article = "<p style='text-align: center'><a href='https://huggingface.co/spaces/breynolds1247/StarryNight_StyleTransfer'>Style Transfer: Create your own stylish paintings </a></p>"
73
 
74
  #examples = [
75
  # ['dog.jpg']
 
44
  probabilities = torch.nn.functional.softmax(output[0], dim=0)
45
 
46
  # Read the categories
 
 
 
47
  categories = {
48
  0:"vanGogh",
49
  1:"Monet",
 
66
  title = "Artist Classifier"
67
  description = "The classifier is a demo classifier to predict the painter using fine-tuned EfficientNet. It allows you to identify the creator of a painting among Van Gogh, Claude Monet, Leonardo da Vinci, Rembrandt, Pablo Picasso, and Salvador Dali. Just upload the image to the left blank box and click the Submit button. A list of confidence will be displayed. Following the link below to find a related work that helps to create your own paintings following the style of painters"
68
 
69
+ article = "<p style='text-align: left'><a href='https://huggingface.co/spaces/breynolds1247/StarryNight_StyleTransfer'>Style Transfer: Create your own stylish paintings </a></p>"
70
 
71
  #examples = [
72
  # ['dog.jpg']