Omnibus commited on
Commit
b97834d
·
1 Parent(s): 8e8e1b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -162,7 +162,7 @@ def compress_data(c,purpose, task, history):
162
  resp = run_gpt(
163
  COMPRESS_DATA_PROMPT,
164
  stop_tokens=["observation:", "task:", "action:", "thought:"],
165
- max_tokens=512,
166
  seed=seed,
167
  purpose=purpose,
168
  task=task,
@@ -178,7 +178,7 @@ def compress_data(c,purpose, task, history):
178
  resp = run_gpt(
179
  COMPRESS_DATA_PROMPT,
180
  stop_tokens=["observation:", "task:", "action:", "thought:"],
181
- max_tokens=512,
182
  seed=seed,
183
  purpose=purpose,
184
  task=task,
@@ -306,11 +306,18 @@ def run(purpose,history):
306
  yield history
307
  if action_name == "COMPLETE":
308
  return history
 
 
 
 
 
 
309
 
310
  gr.ChatInterface(
311
  fn=run,
312
  chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
313
  title="Mixtral 46.7B Powered <br> Huggingface Model Search",
 
314
  ).launch()
315
 
316
 
 
162
  resp = run_gpt(
163
  COMPRESS_DATA_PROMPT,
164
  stop_tokens=["observation:", "task:", "action:", "thought:"],
165
+ max_tokens=1024,
166
  seed=seed,
167
  purpose=purpose,
168
  task=task,
 
178
  resp = run_gpt(
179
  COMPRESS_DATA_PROMPT,
180
  stop_tokens=["observation:", "task:", "action:", "thought:"],
181
+ max_tokens=1024,
182
  seed=seed,
183
  purpose=purpose,
184
  task=task,
 
306
  yield history
307
  if action_name == "COMPLETE":
308
  return history
309
+ examples =[
310
+ "find a model that I can use to generate an image by providing a text prompt",
311
+ "return the top 10 models that I can use to identify objects in images",
312
+ "which models are the most popular right now?"
313
+ ]
314
+
315
 
316
  gr.ChatInterface(
317
  fn=run,
318
  chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
319
  title="Mixtral 46.7B Powered <br> Huggingface Model Search",
320
+ examples=examples,
321
  ).launch()
322
 
323