Uhhy commited on
Commit
1bde1ab
1 Parent(s): 365f24d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import httpx
6
  import asyncio
7
  import gradio as gr
8
  import os
 
9
 
10
  global_data = {
11
  'models': {},
@@ -82,6 +83,7 @@ def remove_duplicates(text):
82
  seen_lines.add(line)
83
  return '\n'.join(unique_lines)
84
 
 
85
  def generate_model_response(model, inputs):
86
  try:
87
  response = model(inputs)
 
6
  import asyncio
7
  import gradio as gr
8
  import os
9
+ from spaces import GPU
10
 
11
  global_data = {
12
  'models': {},
 
83
  seen_lines.add(line)
84
  return '\n'.join(unique_lines)
85
 
86
+ @GPU(duration=60)
87
  def generate_model_response(model, inputs):
88
  try:
89
  response = model(inputs)