Update app.py

app.py

@@ -8,6 +8,8 @@ import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.colors import to_hex
 from bs4 import BeautifulSoup
+from spaces import GPU
+
 
 def clean_html(html_content):
     # Parse the HTML
@@ -286,6 +288,7 @@ def gen_json(input_text, max_new_tokens):
     return
 
 # Creating the Gradio Interface
+@GPU
 def generate_text(input_text, messages=None):
     if input_text == "":
         yield "Please enter some text first."