Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|