Update app.py

app.py

@@ -12,13 +12,14 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 # Load the Lora model
 model = PeftModel.from_pretrained(model, peft_model_id)
 
-model.config.to_json_file("config.json")
 
+@spaces.GPU(duration=200)
 def get_completion(query, model, tokenizer):
     inputs = tokenizer(query, return_tensors="pt")
     outputs = model.generate(**inputs, max_new_tokens=512, do_sample=True, temperature=0.7)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+@spaces.GPU(duration=200)
 def code_review(code_to_analyze):
     query = f"As a code review expert, your role will be to carefully examine the code for potential security flaws and provide guidance on secure coding practices. This may include identifying common coding mistakes that could lead to vulnerabilities, suggesting ways to improve the code's overall security, and recommending tools or techniques that can be used to detect and prevent potential threats. Your expertise in security will be particularly valuable in ensuring that any code developed meets the highest security standard:\n{code_to_analyze}"
     result = get_completion(query, model, tokenizer)