Update app.py
Browse files
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
|