Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import torch
 import tiktoken
 from dataclasses import dataclass
+import torch.nn as nn
+from torch.nn import functional as F
 
 @dataclass
 class GPTConfig: