Rijgersberg commited on
Commit
ef97e39
·
verified ·
1 Parent(s): 70d3fa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,7 +19,7 @@ if torch.cuda.is_available():
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
 
21
 
22
- @spaces.GPU(duration=160)
23
  def generate(
24
  message: str,
25
  chat_history: list[tuple[str, str]],
@@ -135,8 +135,8 @@ Twee van de wortelbedrijven werkten mee door meer informatie over de ongeoorloof
135
  ```
136
  Vat bovenstaand artikel samen"""]
137
  ],
138
- title="google/gemma-2-9b-it",
139
- description="""google/gemma-2-9b-it quick demo""",
140
  submit_btn="Genereer",
141
  stop_btn="Stop",
142
  retry_btn="🔄 Opnieuw",
 
19
  tokenizer = AutoTokenizer.from_pretrained(model_id)
20
 
21
 
22
+ @spaces.GPU(duration=45)
23
  def generate(
24
  message: str,
25
  chat_history: list[tuple[str, str]],
 
135
  ```
136
  Vat bovenstaand artikel samen"""]
137
  ],
138
+ title="meta-llama/Meta-Llama-3.1-8B",
139
+ description="""meta-llama/Meta-Llama-3.1-8B quick demo""",
140
  submit_btn="Genereer",
141
  stop_btn="Stop",
142
  retry_btn="🔄 Opnieuw",