Update app.py

app.py

@@ -12,22 +12,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained(model_repo_path)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_repo_path).to(device)
 
-# Inject CSS for background image
-page_bg_img = '''
-<style>
-body {
-    background-image: url("https://wallpapercave.com/wp/wp5772281.jpg");
-    background-size: cover;
-}
-.block-container {
-    background-color: rgba(255, 255, 255, 0.8);
-    border-radius: 10px;
-    padding: 20px;
-}
-</style>
-'''
-st.markdown(page_bg_img, unsafe_allow_html=True)
-
 # Streamlit app layout
 st.title("Minecraft Query Wizard")
 
@@ -44,3 +28,10 @@ if st.button("Get Answer"):
                 # Generate the answer
                 outputs = model.generate(inputs['input_ids'], max_length=150, num_beams=4, early_stopping=True)
                 # Decode the generated answer
+                answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
+                st.subheader("Answer")
+                st.write(answer)
+            except Exception as e:
+                st.error(f"Error during question answering: {e}")
+    else:
+        st.warning("Please enter a question to get an answer.")