srijaydeshpande commited on
Commit
7a3b3f8
·
verified ·
1 Parent(s): 0c4fe37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ hf_hub_download(
26
  )
27
 
28
  @spaces.GPU(duration=120)
29
- def get_itinerary(llm, information, maxtokens, temperature, top_probability):
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,