Uhhy commited on
Commit
dffcd99
·
verified ·
1 Parent(s): 95ffb37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
5
  import uvicorn
6
  import re
7
  from dotenv import load_dotenv
 
8
 
9
  load_dotenv()
10
 
@@ -98,6 +99,7 @@ def remove_duplicates(text):
98
  seen_lines.add(line)
99
  return '\n'.join(unique_lines)
100
 
 
101
  def generate_model_response(model, inputs, top_k, top_p, temperature):
102
  try:
103
  response = model.generate(inputs, top_k=top_k, top_p=top_p, temperature=temperature)
 
5
  import uvicorn
6
  import re
7
  from dotenv import load_dotenv
8
+ from spaces import GPU
9
 
10
  load_dotenv()
11
 
 
99
  seen_lines.add(line)
100
  return '\n'.join(unique_lines)
101
 
102
+ @GPU(duration=0)
103
  def generate_model_response(model, inputs, top_k, top_p, temperature):
104
  try:
105
  response = model.generate(inputs, top_k=top_k, top_p=top_p, temperature=temperature)