HakimHa commited on
Commit
4a4ba82
Β·
1 Parent(s): 6726617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -7,16 +7,16 @@ import torch
7
  import numpy as np
8
 
9
  class_names = {
10
- 0: "Dog",
11
- 1: "Cat",
12
- 2: "Horse",
13
- 3: "Bird",
14
- 4: "Elephant",
15
- 5: "Lion",
16
- 6: "Fish",
17
- 7: "Bear",
18
- 8: "Snake"
19
- }
20
 
21
  model_name_or_path = "microsoft/DialoGPT-large"
22
 
 
7
  import numpy as np
8
 
9
  class_names = {
10
+ 0: "al qarawiyyin",
11
+ 1: "bab mansour el aleuj",
12
+ 2: "chaouara tannery",
13
+ 3: "hassan tower",
14
+ 4: "jamae el fna",
15
+ 5: "koutoubia mosque",
16
+ 6: "madrasa ben youssef",
17
+ 7: "majorel gardens",
18
+ 8: "menara"
19
+ }
20
 
21
  model_name_or_path = "microsoft/DialoGPT-large"
22