Vitrous commited on
Commit
5d663d2
·
verified ·
1 Parent(s): 6c91a05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = {}