Uhhy commited on
Commit
3488f0a
1 Parent(s): 506360c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import httpx
6
  import asyncio
7
  import gradio as gr
8
  import os
9
- from spaces import GPU
10
  from dotenv import load_dotenv
11
 
12
  load_dotenv()
@@ -88,7 +88,7 @@ def remove_duplicates(text):
88
  seen_lines.add(line)
89
  return '\n'.join(unique_lines)
90
 
91
- @GPU(duration=1)
92
  def generate_model_response(model, inputs):
93
  try:
94
  response = model(inputs)
 
6
  import asyncio
7
  import gradio as gr
8
  import os
9
+ import spaces
10
  from dotenv import load_dotenv
11
 
12
  load_dotenv()
 
88
  seen_lines.add(line)
89
  return '\n'.join(unique_lines)
90
 
91
+ @spaces.GPU(queue=False)
92
  def generate_model_response(model, inputs):
93
  try:
94
  response = model(inputs)