Vitrous commited on
Commit
c33f630
·
verified ·
1 Parent(s): a31cce5

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.from_json("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 = load_dataset("articko/conversations/conversations.jsonl")
20
  # Initialize FastAPI application
21
  app = FastAPI(root_path="/api/v1")
22
  conversations = {}