books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add PyPI package (#576)

Browse Source 
* Add PyPI package

* fixes

* fixes
This commit is contained in:
Sebastian Raschka
2025-03-23 19:28:49 -05:00
committed by GitHub
parent 85f2bc0a58
commit 7114ccd10d
52 changed files with 2173 additions and 2585 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
ch06/04_user_interface/app.py
View File

@@ -10,10 +10,11 @@ import tiktoken
import torch
import chainlit
from previous_chapters import (
classify_review,
GPTModel
)
# For llms_from_scratch installation instructions, see:
# https://github.com/rasbt/LLMs-from-scratch/tree/main/pkg
from llms_from_scratch.ch04 import GPTModel
from llms_from_scratch.ch06 import classify_review
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")