Update app.py

app.py

@@ -2,11 +2,10 @@
 
 import gradio as gr
 import re
-import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 checkpoint = "Pipatpong/vcm_santa"
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cpu"
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True, device_map="auto", load_in_8bit=True)
 
@@ -17,7 +16,8 @@ def generate(text, max_length, num_return_sequences=1):
     return gen_text
 
 
-def extract_functions(text):
+def extract_functions
+(text):
     function_pattern = r'def\s+(\w+)\((.*?)\):([\s\S]*?)return\s+(.*?)\n'
     functions = re.findall(function_pattern, text, flags=re.MULTILINE)
     extracted_text = []