ruhzi commited on
Commit
6a1eaed
·
verified ·
1 Parent(s): 5acdaf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -18,7 +18,7 @@ with open(template_file, "r", encoding="utf-8") as f:
18
 
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_path,
21
- torch_dtype=torch.float32,
22
  low_cpu_mem_usage=True
23
  )
24
 
@@ -84,7 +84,6 @@ demo = gr.ChatInterface(
84
  fn=chat_inference,
85
  title="Indian History SLM",
86
  description="Ask me anything about Indian History!",
87
- type="messages", # Gradio 6.x: enables the dict-based history format
88
  stop_btn="Stop",
89
  concurrency_limit=1,
90
  )
 
18
 
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_path,
21
+ dtype=torch.float32,
22
  low_cpu_mem_usage=True
23
  )
24
 
 
84
  fn=chat_inference,
85
  title="Indian History SLM",
86
  description="Ask me anything about Indian History!",
 
87
  stop_btn="Stop",
88
  concurrency_limit=1,
89
  )