Update app.py

app.py

@@ -18,17 +18,21 @@ model = SentenceTransformer(sentence_similarity_model)
 # Loading in text-generation models
 stoic_generator = pipeline("text-generation", model="eliwill/stoic-generator-10e")
 krishnamurti_generator =  pipeline("text-generation", model="eliwill/distilgpt2-finetuned-final-project")
+alan_generator = pipeline("text-generation", model="eliwill/alan-watts-8e")
+
 
 # Creating philosopher dictionary
 philosopher_dictionary = {
     "Marcus Aurelius": {
         "generator": stoic_generator,
-        "dataframe": stoic_df
+        "dataframe": stoic_df,
+        "image": "marcus-aurelius.jpg"
         },
 
     "Krishnamurti": {
         "generator": krishnamurti_generator,
-        "dataframe": krishnamurti_df
+        "dataframe": krishnamurti_df,
+        "image": "krishnamurti.jpg"
         }
 }
 
@@ -54,7 +58,7 @@ def get_similar_quotes(philosopher, question):
   return top5quotes
 
 def main(question, philosopher):
-  out_image = "marcus-aurelius.jpg"
+  out_image = philosopher_dictionary[philosopher]['image']
   return ask_philosopher(philosopher, question), get_similar_quotes(philosopher, question), out_image
   
 with gr.Blocks(css=".gradio-container {background-image: url('file=blue_mountains.jpg')} # title {color: #F0FFFF}") as demo:
@@ -67,7 +71,7 @@ with gr.Blocks(css=".gradio-container {background-image: url('file=blue_mountain
     with gr.Row():
         with gr.Column():
             inp1 = gr.Textbox(placeholder="Place your question here...", label="Ask a question", elem_id="title")
-            inp2 = gr.Dropdown(choices=["Marcus Aurelius", "Krishnamurti"], value="Marcus Aurelius", label="Choose a philosopher")
+            inp2 = gr.Dropdown(choices=["Marcus Aurelius", "Krishnamurti", "Alan Watts"], value="Marcus Aurelius", label="Choose a philosopher")
           
         out1 = gr.Textbox(
                     lines=3,