antony-pk's picture
Import necessary packages
d6aa4b0 verified
raw
history blame
198 Bytes
import gradio as gr
import torch
from transformers import (
AutoModelForCausalLM,
AutoTokenizer,
TextIteratorStreamer
)
import os
import threading import Thread
import spaces
import time