Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ hf_hub_download(
|
|
26 |
)
|
27 |
|
28 |
@spaces.GPU(duration=120)
|
29 |
-
def get_itinerary(
|
30 |
llm = Llama(
|
31 |
model_path="models/" + model_id,
|
32 |
flash_attn=True,
|
|
|
26 |
)
|
27 |
|
28 |
@spaces.GPU(duration=120)
|
29 |
+
def get_itinerary(information, maxtokens=1000, temperature=0.9, top_probability=0.9):
|
30 |
llm = Llama(
|
31 |
model_path="models/" + model_id,
|
32 |
flash_attn=True,
|