Update app.py

app.py

@@ -5,8 +5,10 @@ import gradio as gr
 def getClassName(fileName):
     fileName = fileName[:-4] # remove extension .jpg
     names = fileName.split('_')
-    artist = ''.join(name+" " for name in names if name[0] in string.ascii_letters)
+    if(names[0] == "Albrecht"):
+        return "Albrecht Dürer"
     
+    artist = ''.join(name+" " for name in names if name[0] in string.ascii_letters)
     return artist[:-1] #Remove last space
 
 def predict(img):