rafaaa2105 commited on
Commit
5f9eae1
·
verified ·
1 Parent(s): f7b7f18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import sys
4
  import asyncio
5
  import transformers
6
  from typing import Any, Dict, List, Union
 
7
 
8
  import chainlit as cl
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
4
  import asyncio
5
  import transformers
6
  from typing import Any, Dict, List, Union
7
+ import torch
8
 
9
  import chainlit as cl
10
  from transformers import AutoTokenizer, AutoModelForCausalLM