salomonsky commited on
Commit
0299673
·
verified ·
1 Parent(s): 679a505

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,7 +4,9 @@ import subprocess
4
  from gtts import gTTS
5
  import cv2
6
  from huggingface_hub import InferenceClient
 
7
 
 
8
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
9
 
10
  def generate_output(prompt):
 
4
  from gtts import gTTS
5
  import cv2
6
  from huggingface_hub import InferenceClient
7
+ import torch
8
 
9
+ device = torch.device("cpu")
10
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
11
 
12
  def generate_output(prompt):