Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -101,7 +101,7 @@ def build_model(model_choice, temperature=0.7):
|
|
| 101 |
if model_choice == "Flan-T5 (Free)":
|
| 102 |
llm = HuggingFaceEndpoint(
|
| 103 |
repo_id="google/flan-t5-base",
|
| 104 |
-
|
| 105 |
task="text2text-generation",
|
| 106 |
max_new_tokens=500,
|
| 107 |
temperature=temperature
|
|
|
|
| 101 |
if model_choice == "Flan-T5 (Free)":
|
| 102 |
llm = HuggingFaceEndpoint(
|
| 103 |
repo_id="google/flan-t5-base",
|
| 104 |
+
huggingfacehub_api_token=api_key,
|
| 105 |
task="text2text-generation",
|
| 106 |
max_new_tokens=500,
|
| 107 |
temperature=temperature
|