Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
|
|
16 |
torch.cuda.empty_cache()
|
17 |
torch.cuda.set_per_process_memory_fraction(0.8) # Adjust the fraction as needed
|
18 |
# Assuming you've created the file conversations.jsonl
|
19 |
-
dataset = Dataset("articko/conversations/conversations.jsonl")
|
20 |
# Initialize FastAPI application
|
21 |
app = FastAPI(root_path="/api/v1")
|
22 |
conversations = {}
|
|
|
16 |
torch.cuda.empty_cache()
|
17 |
torch.cuda.set_per_process_memory_fraction(0.8) # Adjust the fraction as needed
|
18 |
# Assuming you've created the file conversations.jsonl
|
19 |
+
dataset = Dataset('json', data_files="articko/conversations/conversations.jsonl")
|
20 |
# Initialize FastAPI application
|
21 |
app = FastAPI(root_path="/api/v1")
|
22 |
conversations = {}
|