Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,6 +45,7 @@ if uploaded_file:
|
|
| 45 |
llm = HuggingFaceHub(
|
| 46 |
repo_id="google/flan-t5-small", # smaller than base
|
| 47 |
huggingfacehub_api_token=HF_TOKEN,
|
|
|
|
| 48 |
model_kwargs={"temperature":0, "max_length":256}
|
| 49 |
)
|
| 50 |
|
|
|
|
| 45 |
llm = HuggingFaceHub(
|
| 46 |
repo_id="google/flan-t5-small", # smaller than base
|
| 47 |
huggingfacehub_api_token=HF_TOKEN,
|
| 48 |
+
task="text2text-generation",
|
| 49 |
model_kwargs={"temperature":0, "max_length":256}
|
| 50 |
)
|
| 51 |
|