Update app.py

app.py

@@ -67,7 +67,7 @@ def validate_data(text, model_name, topic):
     if model_name not in ['GPT2XL', 'PHI2']:
         return f'Model {model_name} not supported'
     
-    if topic not in ['Characters', 'Locations', 'Nature', 'Video games', 'Series', 'Movies', 'War']:
+    if check_topic(topic) == None:
         return f'Topic {topic} not supported'
     
     return ''
@@ -82,4 +82,4 @@ def check_topic(topic):
         'wars': 'war'
     }
 
-    return topic_dict[topic]
+    return topic_dict[topic] if topic in topic_dict else None