Update app.py

app.py

@@ -1,18 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-#1. Load the AI Brain (We are using a free one called GPT-2)
-print("Loading AI...")
-generator = pipeline('text-generation', model='gpt2')
+MODEL_NAME = "gpt2"
 
-#2 Define the function (what happens when you ask it)
-def ask_brain(question);
-#this tells the AI to generate an answer
-result = generator(question, max_length= 100, num_return_sequences=1)
-answer = result[0]
-['generated_text']
-#we return the answer to the user return answer 
+print(f"Loading{MODEL_NAME}...")
+generator = pipeline('text-generation', model = MODEL_NAME)
 
-#3 Launch the app (this cerates the API)
-iface = gr.Interface(fn=ask_brain, inputs="text", output="text")
-iface.launch()
+def ask_brain(question):
+    result = generator(question, max_length = 100, num_return_sequences=1)
+    answer= result[0]
+    ['generated_text']
+    return answer.replace(question, "").strip()
+
+    iface = gr.Interface(fn=ask_brain, inputs="text", outputs="text"
+                        )
+    iface.launch()