Spaces:

Garvitj
/

emotion-llm

Sleeping

Garvitj commited on Jan 18

Commit

25e2b6c

verified ·

1 Parent(s): e91c92a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -282,7 +282,7 @@ def transcribe_and_predict_video(video, chat_history=[]):
     Bot:"""
     # Tokenize input
-    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     # Generate response
     output = model.generate(**inputs, max_length=512, temperature=0.7, top_p=0.9, do_sample=True)

     Bot:"""
     # Tokenize input
+    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
     # Generate response
     output = model.generate(**inputs, max_length=512, temperature=0.7, top_p=0.9, do_sample=True)