Spaces:
Running
Running
Commit
•
3af16c6
1
Parent(s):
2eca8bb
Update app.py
Browse files
app.py
CHANGED
@@ -2,12 +2,12 @@ from langchain import HuggingFaceHub
|
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
|
5 |
-
|
6 |
|
7 |
checkpoint = "Mr-Vicky-01/Bart-Finetuned-conversational-summarization"
|
8 |
|
9 |
model = HuggingFaceHub(repo_id=checkpoint,
|
10 |
-
huggingfacehub_api_token = os.getenv("HF_TOKEN"),
|
11 |
model_kwargs={"temperature":0.1,
|
12 |
"max_new_tokens":100,
|
13 |
"do_sample":False
|
|
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
|
5 |
+
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv("HF_TOKEN")
|
6 |
|
7 |
checkpoint = "Mr-Vicky-01/Bart-Finetuned-conversational-summarization"
|
8 |
|
9 |
model = HuggingFaceHub(repo_id=checkpoint,
|
10 |
+
# huggingfacehub_api_token = os.getenv("HF_TOKEN"),
|
11 |
model_kwargs={"temperature":0.1,
|
12 |
"max_new_tokens":100,
|
13 |
"do_sample":False
|