Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
import os
|
| 3 |
import time
|
| 4 |
import random
|
|
@@ -140,4 +140,4 @@ def generate_summary(req: SummaryRequest):
|
|
| 140 |
res = ollama.chat(model=OLLAMA_MODEL, messages=[{'role': 'user', 'content': prompt}])
|
| 141 |
return {"summary": res['message']['content'].split("</think>")[-1].strip().replace('"','')}
|
| 142 |
except: return {"summary": "Good job."}
|
| 143 |
-
|
|
|
|
| 1 |
+
|
| 2 |
import os
|
| 3 |
import time
|
| 4 |
import random
|
|
|
|
| 140 |
res = ollama.chat(model=OLLAMA_MODEL, messages=[{'role': 'user', 'content': prompt}])
|
| 141 |
return {"summary": res['message']['content'].split("</think>")[-1].strip().replace('"','')}
|
| 142 |
except: return {"summary": "Good job."}
|
| 143 |
+
|