Update app.py

app.py

@@ -42,8 +42,7 @@ end_idx = torch.argmax(end_logits) + 1
 answer_span = inputs["input_ids"][0][start_idx:end_idx]
 answer = tokenizer.decode(answer_span)
 
-
-#if len(question) > 5:
-st.write(f"Answer: {answer}")
-#else:
-    #st.write('Please ask a longer question about the Amazon')
+try:
+    st.write(f"Answer: {answer}")
+except:
+    st.write('Please ask a longer question about the Amazon')