Omnibus commited on
Commit
d1cbc77
1 Parent(s): 05da1a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def test_models():
26
  try:
27
  print(f'trying: {model}\n')
28
  client= InferenceClient(model)
29
- outp=client.text_generation("What is a cat")
30
  time_delta = datetime.now() - start_time
31
  count=time_delta.total_seconds()
32
  #if time_delta.total_seconds() >= 180:
 
26
  try:
27
  print(f'trying: {model}\n')
28
  client= InferenceClient(model)
29
+ outp=client.text_generation("What is a cat",details=True, return_full_text=True)
30
  time_delta = datetime.now() - start_time
31
  count=time_delta.total_seconds()
32
  #if time_delta.total_seconds() >= 180: