Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
 import gradio as gr
 import torch
+import spaces
 
 model_id = "deepapaikar/Llama_SCplusQA_10epochs"
 print("Before loading model")
@@ -27,7 +28,7 @@ pipe = pipeline(
     model=model,
     tokenizer=tokenizer,
     max_length=100,
-    trust_remote_code=True
+    trust_remote_code=True,
 )
 print("Model loaded successfully")
 #system_message = "Answer the questions truthfully and to the point."