Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -11,15 +11,15 @@ import re
|
|
11 |
|
12 |
# Ensure Playwright installs required browsers and dependencies
|
13 |
subprocess.run(["playwright", "install"])
|
14 |
-
subprocess.run(["playwright", "install-deps"])
|
15 |
|
16 |
# Load environment variables
|
17 |
load_dotenv()
|
18 |
HUGGINGFACEHUB_API_TOKEN = os.getenv('HF_TOKEN')
|
19 |
|
20 |
# Initialize the model instances
|
21 |
-
|
22 |
-
repo_id = "Qwen/Qwen2.5-72B-Instruct"
|
23 |
|
24 |
llm_model_instance = HuggingFaceEndpoint(
|
25 |
repo_id=repo_id, max_length=128, temperature=0.5, token=HUGGINGFACEHUB_API_TOKEN
|
|
|
11 |
|
12 |
# Ensure Playwright installs required browsers and dependencies
|
13 |
subprocess.run(["playwright", "install"])
|
14 |
+
# subprocess.run(["playwright", "install-deps"])
|
15 |
|
16 |
# Load environment variables
|
17 |
load_dotenv()
|
18 |
HUGGINGFACEHUB_API_TOKEN = os.getenv('HF_TOKEN')
|
19 |
|
20 |
# Initialize the model instances
|
21 |
+
repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
|
22 |
+
# repo_id = "Qwen/Qwen2.5-72B-Instruct"
|
23 |
|
24 |
llm_model_instance = HuggingFaceEndpoint(
|
25 |
repo_id=repo_id, max_length=128, temperature=0.5, token=HUGGINGFACEHUB_API_TOKEN
|