Update app_hf.py

app_hf.py

@@ -114,7 +114,8 @@ def init_vector_store():
     return vector_store
 
 def generate_answer(query: str, context: str) -> str:
-    prompt      generator = pipeline('text-generation', model='gpt2')
+    try:
+        generator = pipeline('text-generation', model='gpt2')
         prompt = f"Context: {prompt}"
         response = generator(prompt, max_length=150, num_return_sequences=1)
         return response[0]['generated_text']