Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-import spaces
+# import spaces
 from collections.abc import Iterator
 from threading import Thread
 import gradio as gr
@@ -20,7 +20,7 @@ DESCRIPTION = """\
 
 # if torch.cuda.is_available():
 model_id = "ai4bharat/IndicTrans3-beta"
-model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto", offload_folder="offload")
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
@@ -49,9 +49,6 @@ LANGUAGES = {
 }
 
 
-# def translate(src_lang, text, tgt_lang):
-    
-#     return "Translation output will appear here..."
 
 @spaces.GPU
 def generate(
@@ -93,7 +90,6 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
-
 def store_feedback(rating, feedback_text):
     if not rating:
         gr.Warning("Please select a rating before submitting feedback.", duration=5)
@@ -123,7 +119,7 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                 
                 text_input = gr.Textbox(
                     placeholder="Enter text to translate...", 
-                    label="", 
+                    label="Input text", 
                     lines=10,
                     max_lines=100,
                     elem_id="input-text"
@@ -206,4 +202,4 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                 outputs=feedback_result
             )
 
-demo.launch()
+demo.launch()