Spaces:
Paused
Paused
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)
|