Spaces:
Sleeping
Sleeping
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -129,15 +129,13 @@ except Exception as e:
|
|
| 129 |
print(f"Error loading recommended models: {str(e)}")
|
| 130 |
print("Falling back to local model")
|
| 131 |
|
| 132 |
-
|
| 133 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
| 134 |
-
model = AutoModelForSequenceClassification.from_pretrained(
|
| 135 |
|
| 136 |
# Load the local categories
|
| 137 |
try:
|
| 138 |
-
|
| 139 |
-
categories_path = hf_hub_download(repo_id=model_id, filename="categories.json")
|
| 140 |
-
with open(categories_path, "r") as f:
|
| 141 |
categories = json.load(f)
|
| 142 |
except Exception as e:
|
| 143 |
print(f"Error loading categories: {str(e)}")
|
|
|
|
| 129 |
print(f"Error loading recommended models: {str(e)}")
|
| 130 |
print("Falling back to local model")
|
| 131 |
|
| 132 |
+
model_path = os.path.dirname(os.path.abspath(__file__))
|
| 133 |
+
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 134 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
| 135 |
|
| 136 |
# Load the local categories
|
| 137 |
try:
|
| 138 |
+
with open(os.path.join(model_path, "categories.json"), "r") as f:
|
|
|
|
|
|
|
| 139 |
categories = json.load(f)
|
| 140 |
except Exception as e:
|
| 141 |
print(f"Error loading categories: {str(e)}")
|