Update app.py

app.py

@@ -1,3 +1,4 @@
+import torch  # Explicitly import torch to avoid import issues
 from transformers import MBartForConditionalGeneration, MBart50Tokenizer, AutoModelForCausalLM, AutoTokenizer, pipeline
 import gradio as gr
 import requests
@@ -20,7 +21,7 @@ tokenizer = MBart50Tokenizer.from_pretrained(translation_model_name)
 translation_model = MBartForConditionalGeneration.from_pretrained(translation_model_name)
 
 # Load a text generation model from Hugging Face
-text_generation_model_name = "EleutherAI/gpt-neo-2.7B"  # Use "EleutherAI/gpt-j-6B" for better quality
+text_generation_model_name = "EleutherAI/gpt-neo-2.7B"  # You can switch to "EleutherAI/gpt-j-6B" if available
 text_tokenizer = AutoTokenizer.from_pretrained(text_generation_model_name)
 text_model = AutoModelForCausalLM.from_pretrained(text_generation_model_name, device_map="auto", torch_dtype=torch.float32)