limitedonly41 commited on
Commit
e354a1c
·
verified ·
1 Parent(s): 0abc3fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -18,7 +18,9 @@ load_in_4bit = True # Use 4-bit quantization to reduce memory usage
18
 
19
  peft_model_name = "limitedonly41/website_qwen2_7b_2"
20
 
21
-
 
 
22
 
23
  def fetch_data(url):
24
  headers = {
@@ -96,6 +98,7 @@ def main(urls):
96
  results.append(result)
97
  return results
98
 
 
99
  @spaces.GPU()
100
  def classify_website(url):
101
  from unsloth import FastLanguageModel # Import moved to the top for model loading
 
18
 
19
  peft_model_name = "limitedonly41/website_qwen2_7b_2"
20
 
21
+ # Initialize model and tokenizer variables
22
+ model = None
23
+ tokenizer = None
24
 
25
  def fetch_data(url):
26
  headers = {
 
98
  results.append(result)
99
  return results
100
 
101
+
102
  @spaces.GPU()
103
  def classify_website(url):
104
  from unsloth import FastLanguageModel # Import moved to the top for model loading