Uhhy commited on
Commit
af7f495
1 Parent(s): 52dd1e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,10 +6,10 @@ 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()
13
 
14
  HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
15
 
@@ -88,7 +88,7 @@ def remove_duplicates(text):
88
  seen_lines.add(line)
89
  return '\n'.join(unique_lines)
90
 
91
- @GPU(duration=5)
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
+ from spaces import GPU
10
  from dotenv import load_dotenv
11
 
12
+ load_dotenv()
13
 
14
  HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
15
 
 
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)