as-cle-bert commited on
Commit
4e854f3
·
verified ·
1 Parent(s): 0fe190c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,7 +28,7 @@ def rewrite_csv_ordered_by_winning_rate(csv_path):
28
  # Save the sorted DataFrame to a new CSV file
29
  df_sorted.to_csv(csv_path, index=False)
30
 
31
- @spaces.GPU(duration=200)
32
  def run_inference(pipe, prompt):
33
  response = pipe(prompt)
34
  bot_message = response[0]["generated_text"]
@@ -50,7 +50,7 @@ def modelB_button():
50
  df.to_csv("models.csv")
51
  rewrite_csv_ordered_by_winning_rate("models.csv")
52
 
53
- @spaces.GPU(duration=200)
54
  def reply(modelA, modelB, prompt):
55
  global choice
56
  choice["ModelA"] = modelA
 
28
  # Save the sorted DataFrame to a new CSV file
29
  df_sorted.to_csv(csv_path, index=False)
30
 
31
+ @spaces.GPU()
32
  def run_inference(pipe, prompt):
33
  response = pipe(prompt)
34
  bot_message = response[0]["generated_text"]
 
50
  df.to_csv("models.csv")
51
  rewrite_csv_ordered_by_winning_rate("models.csv")
52
 
53
+ @spaces.GPU()
54
  def reply(modelA, modelB, prompt):
55
  global choice
56
  choice["ModelA"] = modelA