Gabriel C commited on
Commit
596d25f
1 Parent(s): 7c07f39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import os
2
 
3
  import gradio as gr
4
  import numpy as np
 
5
  from groq import Groq
6
  from transformers import pipeline
7
  from transformers.utils import is_flash_attn_2_available
 
2
 
3
  import gradio as gr
4
  import numpy as np
5
+ import torch
6
  from groq import Groq
7
  from transformers import pipeline
8
  from transformers.utils import is_flash_attn_2_available