Update app.py

app.py

@@ -1,4 +1,5 @@
 # Imports
+import spaces
 import os
 import streamlit as st
 import requests
@@ -79,6 +80,7 @@ def main():
         st.image(uploaded_file, caption='🖼️ Uploaded Image', use_column_width=True)
 
         # Initiates AI processing and story generation
+        @spaces.GPU
         with st.spinner("## 🤖 AI is at Work! "):
             scenario = img2txt("uploaded_image.jpg")  # Extracts text from the image
             story = txt2story(scenario, top_k, top_p, temperature)  # Generates a story based on the image text, LLM params