Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import os
 from dotenv import load_dotenv
+import spaces  # Import the spaces module
 
 # Load environment variables from .env file
 load_dotenv()
@@ -45,7 +46,8 @@ system_instruction = (
     "The scripts must be functional in a real-world scenario."
 )
 
-# Define the chat function
+# Define the chat function and decorate it with @spaces.GPU
+@spaces.GPU
 def chat_function(prompt):
     # Create the full input prompt including the system instruction
     full_prompt = f"{system_instruction}\nUser: {prompt}\nAssistant:"
@@ -69,8 +71,8 @@ iface = gr.Interface(
     fn=chat_function,
     inputs="text",
     outputs="text",
-    title="Hacker by Data-Ject Solutions",
-    description="A HackingGPT created by Data-Ject Solutions",
+    title="Hacker",
+    description="A HackingGPT",
 )
 
 # Launch the interface