Update app.py

app.py

@@ -12,6 +12,22 @@ 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")
 
@@ -28,10 +44,3 @@ 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.")